Training and deploying machine learning models on Amazon SageMaker
FlashInfer: Kernel Library for LLM Serving
The official Python client for the Huggingface Hub
A high-throughput and memory-efficient inference and serving engine
DoWhy is a Python library for causal inference
Adversarial Robustness Toolbox (ART) - Python Library for ML security
AIMET is a library that provides advanced quantization and compression
A lightweight vision library for performing large object detection
Bring the notion of Model-as-a-Service to life
Data manipulation and transformation for audio signal processing
Pytorch domain library for recommendation systems
Lightweight Python library for adding real-time multi-object tracking
State-of-the-art diffusion models for image and audio generation
Library for OCR-related tasks powered by Deep Learning
Probabilistic reasoning and statistical analysis in TensorFlow
A Unified Library for Parameter-Efficient Learning
PyTorch extensions for fast R&D prototyping and Kaggle farming
Official inference library for Mistral models
Library for serving Transformers models on Amazon SageMaker
A unified framework for scalable computing
A library for accelerating Transformer models on NVIDIA GPUs
MII makes low-latency and high-throughput inference possible
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
INT4/INT5/INT8 and FP16 inference on CPU for RWKV language model
A GPU-accelerated library containing highly optimized building blocks