Differentiable Simulation Python Packages

sionna

Sionna: An Open-Source Library for Research on Communication Systems

sionna-rt

Sionna RT: The Ray Tracing Package of Sionna

sionna-no-rt

Sionna: An Open-Source Library for Research on Communication Systems

cuwave

Single-GPU, differentiable higher-order finite difference wave propagation in Python: intended for inverse problems and structural optimization, e.g., full waveform inversion and transient topology optimization

pytorchfire

GPU-accelerated wildfire simulator with differentiable cellular automata in PyTorch, supporting real-time parameter calibration via gradient descent

rfx-fdtd

JAX-based 3D FDTD electromagnetic simulator

softjax

A library for soft differentiable relaxations of common JAX functions.

jaxonomy

JAX-native simulation engine for hybrid dynamical systems — block-diagram modeling, ODE/DAE solvers, events, acausal physics, and end-to-end autodiff (grad/vmap/jit).

jaxterity

JAX-native robotics: URDF/MJCF import, differentiable MJX dynamics, system identification, calibrated actuators & sensors, whole-body control, and per-unit attested digital twins — one calibrated model from sim to silicon.

maddening

Modular Automatic Differentiation and Data-Enhanced Neural-network INteracting Graph (MADDENING). A pure JAX end-to-end autodifferentiable graph based framework for multiphysics simulation.

diffpssi

A Python based framework for differentiable power system simulation and dynamic optimization

rewarped

Reinforcement learning in differentiable multiphysics simulation with NVIDIA Warp.

mime-engine

MIcrorobotics Multiphysics Engine (MIME), Differentiable physics for medical microrobots (currently focused on magnetically actuated and in confined biological flows). Stokeslet BEM, IB-LBM, and DiFVM inspired FVM-IBM (with optional GNN correction); built on MADDENING.