Superduper: Integrate AI models and machine learning workflows
A lightweight vision library for performing large object detection
Library for serving Transformers models on Amazon SageMaker
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
State-of-the-art diffusion models for image and audio generation
Create HTML profiling reports from pandas DataFrame objects
Uplift modeling and causal inference with machine learning algorithms
DoWhy is a Python library for causal inference
A high-performance ML model serving framework, offers dynamic batching
Unified Model Serving Framework
Deep learning optimization library: makes distributed training easy
Low-latency REST API for serving text-embeddings
Trainable models and NN optimization tools
Integrate, train and manage any AI models and APIs with your database
Pytorch domain library for recommendation systems
PyTorch extensions for fast R&D prototyping and Kaggle farming
Lightweight Python library for adding real-time multi-object tracking
Multi-Modal Neural Networks for Semantic Search, based on Mid-Fusion
Tensor search for humans
A set of Docker images for training and serving models in TensorFlow
AIMET is a library that provides advanced quantization and compression
Powering Amazon custom machine learning chips
PyTorch library of curated Transformer models and their components
A library to communicate with ChatGPT, Claude, Copilot, Gemini
An easy-to-use LLMs quantization package with user-friendly apis