Projects in Awesome Lists tagged with diffrax
A curated list of projects in awesome lists tagged with diffrax .
https://github.com/proximafusion/raytrax
Raytrax: ECRH ray tracing in JAX
diffrax fusion jax plasma-physics python pyvista
Last synced: 11 May 2026
https://github.com/GalacticDynamics/diffraxtra
Extras for Diffrax: OOP and vectorization
deep-learning differential-equations diffrax dynamical-systems equinox jax machine-learning neural-differential-equations neural-networks
Last synced: 22 Jun 2026
https://github.com/marvinsxtr/jax-flow-matching
2D Flow Matching in JAX with equinox and diffrax
diffrax educational equinox flow-matching jax
Last synced: 06 Oct 2025
https://github.com/galacticdynamics/diffraxtra
Extras for Diffrax: OOP and vectorization
deep-learning differential-equations diffrax dynamical-systems equinox jax machine-learning neural-differential-equations neural-networks
Last synced: 27 Oct 2025