92 dependents
| Package | Description | Downloads/month |
|---|---|---|
| High-performance quantum systems simulation with JAX (GPU-accelerated & differen... | 14K | |
| Stellarator Equilibrium and Optimization Suite | 12K | |
| Normalizing-flow enhanced sampling package for probabilistic inference in JAX | 10K | |
| A general Python-based successive convexification implementation which uses a JA... | 10K | |
| JAX backend for QuTiP | 5K | |
| Bayesian Inference with JAX | 5K | |
| Collimator.ai core simulation engine and API client | 3K | |
| Python+JAX package for differentiable CMB+BBN computations. See https://abcmb.re... | 3K | |
| Gyrokinetic Spectral Solver of Plasma Turbulence | 3K | |
| Circulax is a differentiable circuit simulation framework built on JAX, Optimist... | 2K | |
| JAX-powered brain network simulation with gradient-based optimization | 2K | |
| A Jaxley-based library of ion channels and syanpses for biophysical neuron model... | 2K | |
| Multi-omic single-cell optimal transport tools | 2K | |
| Library for simulation with mathematical epidemiology models | 2K | |
| A Python package for linear subspace identification, nonlinear system identifica... | 2K | |
| Simulation-based inference in JAX | 2K | |
| A multivariate RNA Velocity model to estimate future cell states with uncertaint... | 2K | |
| Continuous-discrete dynamical systems with JAX and related libraries. | 1K | |
| A Jax-native control theory library | 1K | |
| Poscidyn (Python oscillator dynamics) is a Python toolkit based on JAX, designed... | 1K | |
| Score-based Diffusion models in JAX. | 1K | |
| Clearbox AI's all-in-one solution for generation and evaluation of synthetic tab... | 1K | |
| A modern astrodynamics library powered by JAX: differentiate, vectorize, JIT to ... | 1K | |
| Extras for Diffrax: OOP and vectorization | 1K | |
| Opifex Computational Framework - JAX-native platform for scientific machine lear... | 903 | |
| Fast, lightweight and parallelised simulation-based inference in JAX. | 894 | |
| Modeling complex perturbations with CellFlow | 808 | |
| A differentiable, GPU-resident chemical kinetics library using JAX | 775 | |
| Probabilistic modeling of cellular dynamics | 761 | |
| 685 | ||
| An open-source carbon box model implementation built on JAX. | 658 | |
| jax for quantum simulations | 648 | |
| JAX-compatible thermodynamic property calculations. | 643 | |
| Template for Python Projects | 634 | |
| Package for RSNL algorithm for simulation-based inference | 633 | |
| Unified rheological analysis framework with JAX acceleration | 630 | |
| Multiqubit Rydberg Gate Optimizer | 624 | |
| A library for building continuous simulation environments, designed to support s... | 606 | |
| Differentiable microbubble dynamics primitives | 605 | |
| A lightweight library for NURBS curves and surfaces | 583 | |
| The eulerian parameter inference (eulerpi) returns a parameter distribution, whi... | 579 | |
| Reinforcement learning environments for process control applications. | 528 | |
| 520 | ||
| Raytrax: ECRH ray tracing in JAX | 510 | |
| Implementation of physical models for flying sport discs. | 471 | |
| Implementation of Rectified Flows in JAX and Equinox. | 468 | |
| Bijections & normalizing flows with JAX/NNX | 464 | |
| Parameterizable Auto-differentiable Simulators of ocean Trajectories in jAX. | 411 | |
| Tools for the inference of compartmental models | 402 | |
| OneScience是基于先进的深度学习框架打造的科学计算工具包，旨在通过一系列高度集成的组件加速科学研究和技术开发进程。 | 388 |
