Flower: A Friendly Federated Learning Framework
An open source implementation of CLIP
A PyTorch-based Speech Toolkit
Deep learning optimization library making distributed training easy
Detecting silent model failure. NannyML estimates performance
PyTorch extensions for fast R&D prototyping and Kaggle farming
Spatiotemporal Signal Processing with Neural Machine Learning Models
The Triton Inference Server provides an optimized cloud
Data science on data without acquiring a copy
Python package built to ease deep learning on graph
An Easy-to-use, Scalable and High-performance RLHF Framework
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
AIMET is a library that provides advanced quantization and compression
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Lightning fast C++/CUDA neural network framework
Hub of ready-to-use datasets for ML models
Data manipulation and transformation for audio signal processing
Machine Learning Pipelines for Kubeflow
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
An MLOps framework to package, deploy, monitor and manage models
A simple forecasting package
Optax is a gradient processing and optimization library for JAX
Sacred is a tool to help you configure, andorganize IDSIA experiments