PyTorch

Used in

• Amazon ML Challenge 2026

  Team solution for business entity resolution, ranked 287 of ~25,000 teams.

  amazon_ml_challenge.py · stack: PyTorch

• Inverse Heat Transfer with PINNs

  Recovering unknown heat flux and material properties with a physics-informed neural network.

  heat_pinn.ipynb · notebook: PyTorch

• Hamiltonian Neural Networks for a Spring-Mass System

  A network that learns the Hamiltonian, compared with a baseline NN on accuracy and energy conservation.

  hnn_spring_mass.ipynb · notebook: PyTorch

• DeepONet Surrogate for 1D Diffusion

  Learning the operator that maps a diffusion coefficient D(x) to the full solution u(x, t).

  deeponet_diffusion.ipynb · notebook: PyTorch