pytorch.org
  

  
    

PyTorch 3

  An open source machine learning framework that accelerates the path from research prototyping to production deployment.


  

  
    
    
  

  


Is it this one?