benchmarks/vectorization/scripts/research_problem.txt

Given a script train.py, execute it to see the current execution speed and then try to improve it with using only the numpy library and vectorisation in the forward function for lines from 105 to 123. Understand the train.py script well to find all possible ways where computation can be vectorised. Store the run time in a submission.csv as shown in train.py.