Need to debug one program in Pytorch

Job ID: 36881799

Budget: $30 – $250 USD

Freelancers needed to debug a program :

Issue : use of autograd in dimension 2 and 3

I need quick help on a mathematical problem I need to solve through a neural network.
The use of autograd in dimension 2 and 3 is absolutely necessary

This project does not require the program to work on multiple platforms. The focus is solely on debugging and improving the program's output.

In annex, the description of the project + the python program

Please read it carefully before answering
Related categories: Python Mathematics Pytorch