Library for OCR-related tasks powered by Deep Learning
Ready-to-use OCR with 80+ supported languages
A Pythonic framework to simplify AI service building
GPU environment management and cluster orchestration
LLM training code for MosaicML foundation models
Bring the notion of Model-as-a-Service to life
Operating LLMs in production
A library for accelerating Transformer models on NVIDIA GPUs
AIMET is a library that provides advanced quantization and compression
A set of Docker images for training and serving models in TensorFlow
A unified framework for scalable computing
Multi-Modal Neural Networks for Semantic Search, based on Mid-Fusion
Official inference library for Mistral models
PyTorch library of curated Transformer models and their components
MII makes low-latency and high-throughput inference possible
Replace OpenAI GPT with another LLM in your app
Powering Amazon custom machine learning chips
State-of-the-art diffusion models for image and audio generation
DoWhy is a Python library for causal inference
Superduper: Integrate AI models and machine learning workflows
Library for serving Transformers models on Amazon SageMaker
Trainable models and NN optimization tools
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction
A high-performance ML model serving framework, offers dynamic batching
Deep learning optimization library: makes distributed training easy