Serve, optimize and scale PyTorch models in production
☆4,362Aug 6, 2025Updated 7 months ago
Alternatives and similar repositories for serve
Users that are interested in serve are comparing it to the libraries listed below
Sorting:
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆10,426Updated this week
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,926Updated this week
- The easiest way to serve AI apps and models - Build Model Inference APIs, Job queues, LLM apps, Multi-model pipelines, and more!☆8,506Mar 9, 2026Updated last week
- PyTorch elastic training☆728Jun 15, 2022Updated 3 years ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,676Dec 1, 2025Updated 3 months ago
- PyTorch extensions for high performance and large scale training.☆3,403Apr 26, 2025Updated 10 months ago
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,807Updated this week
- The open source AI engineering platform for agents, LLMs, and ML models. MLflow enables teams of all sizes to debug, evaluate, monitor, a…☆24,730Updated this week
- A library for efficient similarity search and clustering of dense vectors.☆39,321Updated this week
- Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.☆41,773Updated this week
- Standardized Distributed Generative and Predictive AI Inference Platform for Scalable, Multi-Framework Deployment on Kubernetes☆5,190Updated this week
- A flexible, high-performance serving system for machine learning models☆6,349Dec 18, 2025Updated 2 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,931Updated this week
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.☆32,191Sep 30, 2025Updated 5 months ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,956Mar 9, 2026Updated last week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,545Updated this week
- An MLOps framework to package, deploy, monitor and manage thousands of production machine learning models☆4,732Mar 1, 2026Updated 2 weeks ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,752Updated this week
- Open standard for machine learning interoperability☆20,484Updated this week
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆19,497Updated this week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆35,038Updated this week
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,642Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,782Updated this week
- Machine Learning Toolkit for Kubernetes☆15,496Jan 5, 2026Updated 2 months ago
- Model interpretability and understanding for PyTorch☆5,572Updated this week
- Production infrastructure for machine learning at scale☆8,028Jun 12, 2024Updated last year
- A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)☆5,622Jan 12, 2026Updated 2 months ago
- An open-source NLP research library, built on PyTorch.☆11,891Nov 22, 2022Updated 3 years ago
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,275Jun 25, 2025Updated 8 months ago
- 🦉 Data Versioning and ML Experiments☆15,448Updated this week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,461Mar 9, 2026Updated last week
- Transformer related optimization, including BERT, GPT☆6,399Mar 27, 2024Updated last year
- Development repository for the Triton language and compiler☆18,656Updated this week
- Ongoing research training transformer models at scale☆15,647Updated this week
- 🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal model…☆157,783Updated this week
- An easy to use PyTorch to TensorRT converter☆4,857Aug 17, 2024Updated last year
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆22,040Jan 23, 2026Updated last month
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆14,344Jul 3, 2024Updated last year
- Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.☆34,198Nov 17, 2025Updated 3 months ago