Embed images and sentences into fixed-length vectors
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
Standalone, small, language-neutral
TFX is an end-to-end platform for deploying production ML pipelines
SkyPilot: Run AI and batch jobs on any infra
Superfast AI decision making and processing of multi-modal data
Unified Model Serving Framework
An MLOps framework to package, deploy, monitor and manage models
A library for scientific machine learning & physics-informed learning
A PyTorch-based Speech Toolkit
Train machine learning models within Docker containers
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
PyTorch extensions for fast R&D prototyping and Kaggle farming
Spatiotemporal Signal Processing with Neural Machine Learning Models
A library for accelerating Transformer models on NVIDIA GPUs
The open-source tool for building high-quality datasets
Hub of ready-to-use datasets for ML models
A library for deep learning end-to-end dialog systems and chatbots
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Machine Learning Systems: Design and Implementation
Optax is a gradient processing and optimization library for JAX
Sacred is a tool to help you configure, andorganize IDSIA experiments
A very simple framework for state-of-the-art NLP