230 dependents
| Package | Description | Downloads/month |
|---|---|---|
| An Open Source Machine Learning Framework for Everyone | 22.4M | |
| Composable transformations of Python+NumPy programs: differentiate, vectorize, J... | 20M | |
| PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial P... | 1.7M | |
| An Open Source Machine Learning Framework for Everyone | 1.4M | |
| Deep universal probabilistic programming with Python and PyTorch | 1.2M | |
| a Jax quantization library | 494K | |
| Python Toolkit for Causal and Probabilistic Reasoning | 448K | |
| An Open Source Machine Learning Framework for Everyone | 377K | |
| An Open Source Machine Learning Framework for Everyone | 305K | |
| Efficient optimizers | 193K | |
| An Open Source Machine Learning Framework for Everyone | 177K | |
| CUDA accelerated equivariant operations | 171K | |
| Quantum program simulators that can run locally | 154K | |
| An Open Source Machine Learning Framework for Everyone | 128K | |
| PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial P... | 77K | |
| Training PyTorch models with differential privacy | 76K | |
| MACE - Fast and accurate machine learning interatomic potentials with higher ord... | 69K | |
| An Open Source Machine Learning Framework for Everyone | 65K | |
| Functional tensors for probabilistic programming | 34K | |
| Functionalities such as a layers for building neural networks in Jax. | 31K | |
| Influence Functions with (Eigenvalue-corrected) Kronecker-factored Approximate C... | 31K | |
| End-to-End Speech Processing Toolkit | 30K | |
| Collection of ML models and benchmarking tools | 26K | |
| Faster simulation of fermionic quantum circuits. | 24K | |
| 18K | ||
| NeuralOperator: Learning in Infinite Dimensions | 18K | |
| Official repository of the xLSTM. | 15K | |
| JAX-based tensor network library with label-based contraction and symmetry-aware... | 15K | |
| TensorFlow is an open source machine learning framework for everyone. | 14K | |
| An Open Source Machine Learning Framework for Everyone | 11K | |
| An Open Source Machine Learning Framework for Everyone | 10K | |
| Stable Pretraining Library for Self-Supervised Learning | 10K | |
| PyTorch compiler that accelerates training and inference. Get built-in optimizat... | 9K | |
| PyTorch Automatic Differentiation Utility for "tad" Projects. | 8K | |
| Abstract algebric structures for GPU-efficient computation | 7K | |
| An Open Source Machine Learning Framework for Everyone | 7K | |
| A tool to find homologous interactions and speed up AlphaFold-based structural m... | 7K | |
| An Open Source Machine Learning Framework for Everyone | 6K | |
| An Open Source Machine Learning Framework for Everyone | 6K | |
| A generic interface for linear algebra backends | 6K | |
| This is the module for detecting and classifying text on rama pictures | 6K | |
| Optimize and simulate measurement-based quantum computation | 6K | |
| Shared utilities and training infrastructure for biomolecular structure predicti... | 6K | |
| PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...) | 6K | |
| Exponential families for JAX | 5K | |
| Gaia - Formal Language for Natural Science | 5K | |
| Scikit-learn compatible package for archetypal analysis. | 4K | |
| An Open Source Machine Learning Framework for Everyone | 4K | |
| Efficient numerical calculation of generalized filter functions | 4K | |
| Laplace approximations for Deep Learning. | 4K |
