50 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Differentiable, Hardware Accelerated, Molecular Dynamics | 95K | |
| Open source code for AlphaFold. | 73K | |
| Biomolecular emulator | 6K | |
| Pipeline allows massive screening using alphafold | 3K | |
| Hybrid ML + physics model of the Earth's atmosphere | 3K | |
| Recurrent Inertial Graph-based Estimator (RING) | 2K | |
| Surjection layers for density estimation with normalizing flows | 2K | |
| Simulation-based inference in JAX | 2K | |
| Gaussian processes in NumPyro and JAX | 1K | |
| juxtapose: Multi-Person Pose Tracking Inference SDK with RTMDet, YOLOv8, GDino, ... | 1K | |
| Finetuning utilities for AlphaGenome with custom heads and parameter freezing | 1K | |
| Structure-based antibody renumbering | 1K | |
| An ai-models plugin to run Deepmind's graphcast model | 895 | |
| Library for efficient training and application of Machine Learning Interatomic P... | 894 | |
| Spyx: Spiking Neural Networks in JAX | 851 | |
| Riemannian JAX | 827 | |
| 810 | ||
| Recourse Explanation Library in JAX | 802 | |
| An ai-models plugin to run Deepmind's graphcast model | 782 | |
| The CLRS Algorithmic Reasoning Benchmark. | 645 | |
| Amortized Inference for Causal Structure Learning, NeurIPS 2022 | 582 | |
| Deep learning quantum Monte Carlo for electrons in real space | 579 | |
| An ai-models plugin to run Deepmind's gencast model | 529 | |
| a modular reinforcement learning library with JAX agents | 450 | |
| Leonardo: an LSFM image processing toolset | 411 | |
| OneScience是基于先进的深度学习框架打造的科学计算工具包，旨在通过一系列高度集成的组件加速科学研究和技术开发进程。 | 388 | |
| Interpretable, protein structure-based prediction of missense variant deleteriou... | 325 | |
| SuperGrad differentiable Hamiltonian simulator | 302 | |
| 🍐 Predicting protein-protein interactions and structures from multiple sequence... | 300 | |
| The implementation for the differentiable model predictive excitation algorithm ... | 293 | |
| Dynamic Programming using Neural Networks | 293 | |
| Serialize JAX, Flax, Haiku, or Objax model params with `safetensors` | 278 | |
| Compiler from RASP to transformer weights | 262 | |
| A simple port of μP to Haiku/JAX. | 236 | |
| 231 | ||
| Training Neural Network potentials through customizable routines in JAX. | 225 | |
| Configurable MLPs built on JAX and Haiku | 224 | |
| LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite | 222 | |
| An advanced neural network framework with interpretability, generalization, robu... | 214 | |
| AlphaFold2 protein predictions | 212 | |
| A tool for creating science experiments in jax, torch, brax, etc | 202 | |
| Interval Analysis and Mixed Monotone Reachability in JAX | 195 | |
| Pretrained AlphaZero agents for Ludax board games | 177 | |
| Reinforcement learning in pure JAX. | 151 | |
| Deep Bayesian unsupervised decoder networks. Use poisson or multinomial belief n... | 106 | |
| Non parametric score function estimation library | 91 | |
| An experiment framework for Haiku and Jax | 91 | |
| Bayesian phase screen inference | 85 | |
| A JAX-based neural network library surpassing Google DeepMind's Haiku and Optax | 81 | |
| Making protein folding accessible in kaggle platform | 77 |
