Hub of ready-to-use datasets for ML models
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
The easiest way to use deep metric learning in your application
AIMET is a library that provides advanced quantization and compression
Detecting silent model failure. NannyML estimates performance
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Training PyTorch models with differential privacy
Open Source Differentiable Computer Vision Library
PyTorch extensions for fast R&D prototyping and Kaggle farming
A lightweight vision library for performing large object detection
Helps scientists define testable, modular, self-documenting dataflow
A Python package to assess and improve fairness of ML models
Build AI-powered semantic search applications
Pytorch domain library for recommendation systems
A library for deep learning end-to-end dialog systems and chatbots
A PyTorch-based Speech Toolkit
Bring the notion of Model-as-a-Service to life
Scalable and user friendly neural forecasting algorithms.
Implementation of 'lightweight' GAN, proposed in ICLR 2021
The most accurate natural language detection library for Python
A simple, performant and scalable Jax LLM
Synthetic data curation for post-training and data extraction
Benchmarking synthetic data generation methods
ktrain is a Python library that makes deep learning AI more accessible
Open Source Computer Vision Library