48 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capab... | 180K | |
| Sequential Least Squares Programming (SLSQP) optimizer implemented in pure JAX | 16K | |
| High-performance quantum systems simulation with JAX (GPU-accelerated & differen... | 14K | |
| Gaussian processes in JAX and Equinox. | 13K | |
| Parametric modeling in JAX + Equinox | 7K | |
| Exponential families for JAX | 5K | |
| Parametric radio frequency modeling, optimization and sampling | 4K | |
| Python+JAX package for differentiable CMB+BBN computations. See https://abcmb.re... | 3K | |
| State of the art inference for your bayesian models. | 2K | |
| JAX-powered brain network simulation with gradient-based optimization | 2K | |
| A Jax-native control theory library | 1K | |
| Constraint-Aware Descent Routine Executor — JAX-native constrained optimization | 1K | |
| Auto-differentiable and hardware-accelerated force density method | 1K | |
| Volatile partitioning between planetary reservoirs with solubility, non-ideality... | 1K | |
| Hartree-Fock solver python package using jax | 919 | |
| Opifex Computational Framework - JAX-native platform for scientific machine lear... | 903 | |
| JAX-native TBATS (forecast::tbats port) — multi-seasonal forecasting with vmap p... | 833 | |
| A library for soft differentiable relaxations of common JAX functions. | 792 | |
| NEural MOdelS, a statistical modeling framework for neuroscience. | 778 | |
| A differentiable, GPU-resident chemical kinetics library using JAX | 775 | |
| Understanding and Predicting Dynamic Choices | 742 | |
| Fast evaluation of SMC' likelihood for population genetic data | 733 | |
| A python-based interactive visualization tool to view XPCS dataset | 696 | |
| JAX-compatible thermodynamic property calculations. | 643 | |
| Cryo-EM Ensemble Optimization using JAX and OpenMM for Molecular Dynamics simula... | 621 | |
| 611 | ||
| A lightweight library for NURBS curves and surfaces | 583 | |
| Infer regulatory modules through informative latent component model in the singl... | 581 | |
| Automated material constitutive modeling in JAX | 502 | |
| Frequency-domain (non)linear state-space identification using JAX. | 492 | |
| This is a Python library for solving Hamilton-Jacobi-Bellman (HJB) equations usi... | 463 | |
| Electromagnetics simulation library for moving point charges built on JAX | 423 | |
| A package for extracting quasinormal modes from time domain data | 387 | |
| Lightweight utility functions for JAX arrays, batching, and pytrees | 386 | |
| Statistical inference library built on JAX pytrees and optimistix optimizers | 333 | |
| High-performance JAX-based filtering for Dynamic Factor Stochastic Volatility (D... | 242 | |
| A JAX-based library for structural modeling and estimation in economics. | 234 | |
| Efficient library for spectral analysis in high-energy astrophysics. | 220 | |
| GPU Powered CMB Parametric Component Seperation using Furax and JAX | 133 | |
| QCM-D data analysis software with rheological modeling | 128 | |
| Python package to fit interferometric data, accelerated using Jax. | 123 | |
| Efficient library for spectral analysis in high-energy astrophysics. | 84 | |
| MESS: Modern Electronic Structure Simulations | 79 | |
| ⚛️ Nonlinear diffusion problems with JAX | 70 | |
| Multistage neural networks with JAX | 64 | |
| A comprehensive toolkit for astronomical time series analysis. | 42 | |
| Multiphysics Topology Optimization | 26 | |
| Statistical Inference with JAX. | 4 |
