140 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Distrax: Probability distributions in JAX. | 206K | |
| Open source code for AlphaFold. | 73K | |
| A library of reinforcement learning building blocks in JAX. | 37K | |
| Functionalities such as a layers for building neural networks in Jax. | 31K | |
| PIX is an image processing library in JAX, for JAX. | 26K | |
| (EasyDel Former) is a utility library designed to simplify and enhance the devel... | 24K | |
| Modular, scalable library to train ML models | 24K | |
| Accelerated, Python-only, single-cell integration benchmarking metrics | 22K | |
| Differentiable Ray Tracing Toolbox for Radio Propagation Simulations | 17K | |
| 17K | ||
| easydel jax kernels writen in triton for gpus and pallas for tpus | 17K | |
| JAX-accelerated meta-reinforcement learning environments inspired by XLand and M... | 16K | |
| Add a tqdm progress bar to your JAX scans and loops. | 16K | |
| Monte Carlo tree search in JAX | 13K | |
| MPAX: Mathematical Programming in JAX | 11K | |
| A diverse suite of scalable reinforcement learning environments in JAX | 9K | |
| JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library | 8K | |
| ⚡ Flashbax: Accelerated Replay Buffers in JAX | 7K | |
| (Crafter + NetHack) in JAX. ICML 2024 Spotlight. | 7K | |
| Multi-Agent Reinforcement Learning with JAX | 6K | |
| Accelerated Quality-Diversity | 4K | |
| (EasyDel Former) is a utility library designed to simplify and enhance the devel... | 4K | |
| A JAX implementation of the Metalog distribution for flexible probability modeli... | 4K | |
| Cost constrained environments with a gymnax interface | 4K | |
| A framework for Alberta Plan research building the foundations of Continual AI. | 3K | |
| Pipeline allows massive screening using alphafold | 3K | |
| Python library for modeling DFAs, Moore Machines, and Transition Systems in JAX. | 3K | |
| A library providing general-purpose tools for estimating discrete distributions ... | 3K | |
| Use bacpipe to streamline the process of generating embeddings and analysing you... | 3K | |
| A library for fast Jax experimentation | 3K | |
| A multiverse of prophet models, for forecasting and Marketing Mix Modeling. | 3K | |
| A JAX-powered derivative of the original gridvoting project for spatial voting s... | 2K | |
| Parabellum environment for parallel warfare simulation | 2K | |
| Hyperbolic Deep Learning in JAX and Flax.nnx | 2K | |
| A multivariate RNA Velocity model to estimate future cell states with uncertaint... | 2K | |
| Continuous-discrete dynamical systems with JAX and related libraries. | 1K | |
| The Bio-inspired Robotics Testbed (BRT) | 1K | |
| Evaluating the Robustness of Time-Series Causal Discovery in Misspecified Scenar... | 1K | |
| Jax-based 2d physics | 1K | |
| The Jansen-Rit neural mass model SDE in JAX | 1K | |
| juxtapose: Multi-Person Pose Tracking Inference SDK with RTMDet, YOLOv8, GDino, ... | 1K | |
| JAX implementation of Reinforcement Learning agents | 1K | |
| 1K | ||
| JAX based ptychography reconstructions through differentiable programming. | 1K | |
| TORAX: Tokamak transport simulation in JAX | 945 | |
| A simple JAX-powered simulation library for numerical experiments of fuzzy dark ... | 913 | |
| A unified framework for implementing and interfacing with MuJoCo and MuJoCo-XLA ... | 903 | |
| A collection of short-horizon gradient-based hyperparameter optimization algorit... | 890 | |
| KKT-HardNet: Physics-constrained neural networks with hard nonlinear equality an... | 873 | |
| Alpa automatically parallelizes large tensor computation graphs and runs them on... | 862 |
