35 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https:... | 208K | |
| Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capab... | 180K | |
| Interpolation and function approximation with JAX | 56K | |
| Optimal transport tools implemented with the JAX framework, to solve large scale... | 28K | |
| Gaussian processes in JAX and Equinox. | 13K | |
| Easy to use distributions, bijections and normalizing flows in JAX. | 7K | |
| Circulax is a differentiable circuit simulation framework built on JAX, Optimist... | 2K | |
| Package for Interpolating Zero Order Intensity Values | 2K | |
| Continuous-discrete dynamical systems with JAX and related libraries. | 1K | |
| A Jax-native control theory library | 1K | |
| Constraint-Aware Descent Routine Executor — JAX-native constrained optimization | 1K | |
| A cryo-electron microscopy image simulation library for building data analysis i... | 1K | |
| Auto-differentiable and hardware-accelerated force density method | 1K | |
| Operators and solvers for high-performance computing. | 985 | |
| Opifex Computational Framework - JAX-native platform for scientific machine lear... | 903 | |
| A library for soft differentiable relaxations of common JAX functions. | 792 | |
| A differentiable, GPU-resident chemical kinetics library using JAX | 775 | |
| JAX-compatible triangular meshes and triangular-mesh-based simulations | 764 | |
| Understanding and Predicting Dynamic Choices | 742 | |
| 🍎 A JAX and Warp library for differentiable physics simulation, featuring elasti... | 624 | |
| Infer regulatory modules through informative latent component model in the singl... | 581 | |
| QUADCOIL, a global stellarator coil optimizer and coil complexity proxy | 509 | |
| Automated material constitutive modeling in JAX | 502 | |
| Parameterizable Auto-differentiable Simulators of ocean Trajectories in jAX. | 411 | |
| RECOVAR cryo-EM heterogeneity analysis | 372 | |
| Engine to compute Jacobian-vector and vector-Jacobian products for (convex) quad... | 344 | |
| Numerical relativity surrogate model for gravitational waveforms in Jax | 332 | |
| Add your description here | 323 | |
| A JAX-based library for structural modeling and estimation in economics. | 234 | |
| Stochastic trace estimation using JAX | 203 | |
| Universal Partial Differential Equations Simulator | 184 | |
| A minimal JAX library for connectivity modelling at scale | 169 | |
| Forecast reconciliation is a post-forecasting process aimed at improving the acc... | 114 | |
| Multistage neural networks with JAX | 64 | |
| Multiphysics Topology Optimization | 26 |
