71 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Massively parallel rigidbody physics simulation on accelerator hardware. | 53K | |
| Optimal transport tools implemented with the JAX framework, to solve large scale... | 28K | |
| Non-parametric density inference for single-cell analysis. | 9K | |
| BlackJAX is a Bayesian Inference library designed for ease of use, speed and mod... | 7K | |
| A high-performance Python package for estimating latent-class conditional logit ... | 3K | |
| Gap-averaged flow simulations with Gaussian Process regression | 3K | |
| Collimator.ai core simulation engine and API client | 3K | |
| State of the art inference for your bayesian models. | 2K | |
| Recurrent Inertial Graph-based Estimator (RING) | 2K | |
| Source code for the ValidMLInference Python Package | 2K | |
| A Python package for linear subspace identification, nonlinear system identifica... | 2K | |
| QArray - a GPU accelerated constant capacitance simulator for quantum dots | 1K | |
| Continuous-discrete dynamical systems with JAX and related libraries. | 1K | |
| Supernova light-curve fitting code | 1K | |
| Gaussian processes in NumPyro and JAX | 1K | |
| Dark matter halo properties and halo mass functions using JAX | 1K | |
| NashOpt - A Python library for computing generalized Nash equilibria and solvin... | 1K | |
| A library for coreset algorithms, written in JAX for fast execution and GPU supp... | 1K | |
| Auto-differentiable and hardware-accelerated force density method | 1K | |
| TORAX: Tokamak transport simulation in JAX | 945 | |
| JAX-powered machine learning and modeling framework for GRaTeR disks. | 865 | |
| Framework for benchmarking optimal transport methods on both synthetic and real ... | 851 | |
| Jax-based quantum state vector and noisy simulator. | 810 | |
| JAX-based scan anomaly detection for time-series residuals | 808 | |
| Importance Sampling for State Space Models (in jax) | 683 | |
| Add your description here | 664 | |
| Differentiable versions of common HEP operations. | 650 | |
| Cryo-EM Ensemble Optimization using JAX and OpenMM for Molecular Dynamics simula... | 621 | |
| Algorithms at the boundary of control and machine learning | 511 | |
| CPU-optimized JAX-based homodyne scattering analysis for XPCS under nonequilibri... | 493 | |
| This is a Python library for solving Hamilton-Jacobi-Bellman (HJB) equations usi... | 463 | |
| JAX package for Generalized Hyperbolic distributions as exponential families | 420 | |
| A differentiable astrophysics simulator in JAX | 420 | |
| An automized PDE solver for accelerators based on JAX | 398 | |
| An Easy Way for Stellar Doppler Imaging of Cool Single Stars. | 398 | |
| Transit signals detection among correlated noises | 375 | |
| RECOVAR cryo-EM heterogeneity analysis | 372 | |
| LEM slope stability analysis using JAX | 334 | |
| Collection of packages aiding in parallelization and optimization of differentia... | 307 | |
| SuperGrad differentiable Hamiltonian simulator | 302 | |
| Classical machine learning using JAX and Equinox | 274 | |
| Machine learning Calabi-Yau metrics | 273 | |
| Joint modeling of multiple deep mutational scanning experiments | 268 | |
| JAX-differentiable AAA algorithm | 259 | |
| differentiable (binned) likelihoods with JAX | 255 | |
| Learn spatially informed Waddington-like potentials for single-cell gene express... | 242 | |
| Kinetic Inverse-Laplace Toolbox for MEM fitting | 241 | |
| Training Neural Network potentials through customizable routines in JAX. | 225 | |
| Diffilqrax is an open-source Python library for differentiable iLQR and LQR in J... | 211 | |
| A JAX-based time-series forecasting library. | 188 |
