151 dependents
| Package | Description | Downloads/month |
|---|---|---|
| An Open Source Machine Learning Framework for Everyone | 22.4M | |
| Deep Learning for humans | 20.9M | |
| Composable transformations of Python+NumPy programs: differentiate, vectorize, J... | 20M | |
| Open standard for machine learning interoperability | 19.3M | |
| Composable transformations of Python+NumPy programs: differentiate, vectorize, J... | 14.7M | |
| Library for reading and writing large multi-dimensional arrays. | 6M | |
| ONNX Script enables developers to naturally author ONNX functions and models usi... | 4.2M | |
| Efficient in-memory representation for ONNX, in Python | 2.6M | |
| Redis Vector Library (RedisVL) -- the AI-native Python client for Redis. | 1.9M | |
| A tool for converting ONNX files to LiteRT/TFLite/TensorFlow, PyTorch native cod... | 1.5M | |
| An Open Source Machine Learning Framework for Everyone | 1.4M | |
| A Toolkit to Help Optimize Onnx Model | 1.1M | |
| NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVID... | 763K | |
| A tile level programming language to generate high performance code. | 475K | |
| An Open Source Machine Learning Framework for Everyone | 377K | |
| An Open Source Machine Learning Framework for Everyone | 305K | |
| Deep Learning for humans | 295K | |
| Open standard for machine learning interoperability | 269K | |
| An Open Source Machine Learning Framework for Everyone | 177K | |
| An Open Source Machine Learning Framework for Everyone | 128K | |
| This API provides programmatic access to the AlphaGenome model developed by Goog... | 89K | |
| TOSA Tools: serialization library, reference model, optional MLIR translator | 68K | |
| An Open Source Machine Learning Framework for Everyone | 65K | |
| AI Edge Quantizer: flexible post training quantization for LiteRT models. | 63K | |
| LiteRT is for mobile and embedded devices. | 45K | |
| export JAX to ONNX | 37K | |
| Easy-to-use and powerful LLM and SLM library with awesome model zoo. | 36K | |
| Multi-backend recommender systems with Keras 3 | 33K | |
| An open-source framework for machine learning and other computations on decentra... | 33K | |
| A retargetable MLIR-based machine learning compiler and runtime toolkit. | 23K | |
| Tensor learning compiler binary distribution | 20K | |
| AI Edge Quantizer: flexible post training quantization for LiteRT models. | 19K | |
| 18K | ||
| 17K | ||
| A deep learning package for many-body potential energy representation and molecu... | 15K | |
| TensorFlow is an open source machine learning framework for everyone. | 14K | |
| For on-the-fly active learning of interatomic potentials | 13K | |
| An Open Source Machine Learning Framework for Everyone | 11K | |
| An Open Source Machine Learning Framework for Everyone | 10K | |
| Python-first Development for AI Compilers | 10K | |
| A JIT compiler for Numpy-based Python programs and bindings for stateful dataflo... | 7K | |
| An Open Source Machine Learning Framework for Everyone | 7K | |
| An Open Source Machine Learning Framework for Everyone | 6K | |
| Single cell trajectory detection | 6K | |
| An Open Source Machine Learning Framework for Everyone | 6K | |
| IREE's PyTorch Frontend, based on Torch Dynamo. | 6K | |
| This is the module for detecting and classifying text on rama pictures | 6K | |
| Wave: Python Domain-Specific Language for High Performance Machine Learning | 5K | |
| Use late-interaction multi-modal models such as ColPali in just a few lines of c... | 5K | |
| Bridging the gap between Statistical Inference and Neural Networks | 5K |