151 dependents
Package Description Downloads/month
An Open Source Machine Learning Framework for Everyone 22.4M
Deep Learning for humans 20.9M
jax-ml jax
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