Differentiable Simulations Python Packages

differt

Differentiable Ray Tracing Toolbox for Radio Propagation Simulations

cheetah-accelerator

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.

pyrddlgym-jax

JAX compilation of RDDL description files, and a differentiable planner in JAX.

phiflow

A differentiable PDE solving framework for machine learning

softtorch

A library for soft differentiable relaxations of common PyTorch functions.

agent-torch

large population models

jwave

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

socialforce

PyTorch implementation of DeepSocialForce.

tensormesh-fem

Differentiable Finite Element Method Library for PyTorch

jaxsso

A differentiable finite element analysis solver for structural optimization based on JAX