Found 43 repositories(showing 30)
patrick-kidger
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
dkweiss31
Quantum optimal control based on dynamiqs, diffrax and JAX
proximafusion
Raytrax: ECRH ray tracing in JAX
cdaunt
Circulax is a differentiable circuit simulation framework built on JAX, Optimistix and Diffrax
forgi86
JAX tutorial for people already familiar with PyTorch
AbeelLab
Building, simulating, and training systems biology models using JAX/Diffrax
MoltenEcdysone09
Gene Regulatory Interaction Network Simulator - GRiNS
GalacticDynamics
Extras for Diffrax: OOP and vectorization
Benjamin-Walker
SBGM
marvinsxtr
2D Flow Matching in JAX with equinox and diffrax
m9h
JAX-accelerated electrophysiology analysis CLI with MNE, Diffrax, and osl-dynamics integration
flatton
No description available
DylanEsguerra
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
aseyboldt
No description available
shappiron
This is the molecular dynamics simulator written with diffrax library.
symmeHub
Kinamax is a JAX/Diffrax powered toolkit for analyzing periodic and subharmonic responses of driven nonlinear ODEs
DiogoFernandes-2008015
Grey Box model for System Identification based on the jax and diffrax packages for a 1RC ECM battery model.
conda-forge
A conda-smithy repository for diffrax.
jbosh1
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
saewoong-oh
No description available
PraneethMerugu
No description available
cokuyucu
Diffraxia is a diffraction data processing toolkit designed for advanced materials characterization workflows (X-ray diffraction, synchrotron data, etc.).
kenneth-meyer
Exploring the diffrax repo
luke-a-thompson
Low-storage Runge-Kutta Methods for Diffrax
Rumcohol
Roating Squares Optimisation in Jax
luke-a-thompson
Geometric Numerical Integrators for Diffrax
LennartGevers
No description available
Trophime
Magnetically Coupled RL Circuit Simulator with diffrax
VMLC-PV
No description available
luke-a-thompson
Rough Differential Equation Integrators for Diffrax