#

diffrax

Here are 10 public repositories matching this topic...

proximafusion / raytrax

Raytrax: ECRH ray tracing in JAX

python fusion plasma-physics jax pyvista diffrax

Updated Aug 3, 2026
Jupyter Notebook

forgi86 / jax-tutorial

JAX tutorial for people already familiar with PyTorch

bayesian-inference ordinary-differential-equations system-identification maml meta-learning hypernetworks jax blackjax diffrax

Updated Jul 15, 2025
Jupyter Notebook

MoltenEcdysone09 / GRiNS

Gene Regulatory Interaction Network Simulator - GRiNS

systems-biology python3 ising-model network-dynamics jax gene-regulatory-networks diffrax parameter-agnostic-simulation

Updated Jan 21, 2026
Python

GalacticDynamics / diffraxtra

Extras for Diffrax: OOP and vectorization

machine-learning deep-learning neural-networks dynamical-systems differential-equations equinox jax neural-differential-equations diffrax

Updated Aug 19, 2026
Python

marvinsxtr / jax-flow-matching

2D Flow Matching in JAX with equinox and diffrax

educational equinox jax flow-matching diffrax

Updated Oct 15, 2025
Jupyter Notebook

DylanEsguerra / Geerts_2023_PBPK_QSP_SBML_Model

An effort to reproduce the Geerts 2022 PBPK/QSP model with SBML and python. References: Geerts H, Walker M, Rose R, et al. A combined physiologically-based pharmacokinetic and quantitative systems pharmacology model for modeling amyloid aggregation in Alzheimer's disease. CPT Pharmacometrics Syst Pharmacol. 2023; 12: 444-461. doi:10.1002/psp4.1291

qsp alzheimers-disease sbml pbpk diffrax

Updated Jul 24, 2025
Python

luke-a-thompson / diffrax-lowstorage

Low-storage Runge-Kutta Methods for Diffrax

numerical-analysis runge-kutta-methods jax diffrax

Updated May 22, 2026
Python

luke-a-thompson / georax

Geometric Numerical Integration for Diffrax.

numerical-analysis jax diffrax manifold-constrained

Updated Aug 4, 2026
Python

esmaeelMhd / digital-twin-engine

Physics-informed latent neural SDEs for surrogate modelling of industrial process systems, in JAX.

uncertainty-quantification equinox process-engineering digital-twin jax scientific-machine-learning surrogate-model physics-informed neural-sde diffrax

Updated Sep 3, 2026
Python

luke-a-thompson / roughrax

Rough Differential Equation Integrators for Diffrax

numerical-analysis jax rough-paths diffrax

Updated Aug 10, 2026
Python

Add this topic to your repo

To associate your repository with the diffrax topic, visit your repo's landing page and select "manage topics."