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

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.

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.

softjax

A library for soft differentiable relaxations of common JAX functions.

rewarped

A platform for reinforcement learning in differentiable multiphysics simulation

diffpssi

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