A package for the sparse identification of nonlinear dynamical systems
Numerical differential equation solvers in JAX
Open-source deep-learning framework for building and training
A library for scientific machine learning & physics-informed learning
Optax is a gradient processing and optimization library for JAX
A PyTorch library for implementing flow matching algorithms
Qwen2.5-Coder is the code version of Qwen2.5, the large language model