Extras for Diffrax: OOP and vectorization
Raytrax: ECRH ray tracing in JAX
Gene Regulatory Interaction Network Simulator - GRiNS
