My PyTorch example could help you.

  
      
      GitHub
  
  
    

Tony-Y/PotFit 76

A PyTorch Example of Potential Fitting . Contribute to Tony-Y/PotFit development by creating an account on GitHub.