Data manipulation and transformation for audio signal processing
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
NVIDIA Federated Learning Application Runtime Environment
Machine learning metrics for distributed, scalable PyTorch application
Detecting silent model failure. NannyML estimates performance
A lightweight vision library for performing large object detection
Optax is a gradient processing and optimization library for JAX
DoWhy is a Python library for causal inference
A simple but complete full-attention transformer
Multilingual Automatic Speech Recognition with word-level timestamps
Algorithmic Trading in Python with Machine Learning
A system for quickly generating training data with weak supervision
A Python Automated Machine Learning tool that optimizes ML
Modular quant framework
Deep learning optimization library making distributed training easy
Decentralized deep learning in PyTorch. Built to train models
Library for OCR-related tasks powered by Deep Learning
Graph Neural Network Library for PyTorch
A library for scientific machine learning & physics-informed learning
PyTorch extensions for fast R&D prototyping and Kaggle farming
Tool for visualizing and tracking your machine learning experiments
The most intuitive, flexible, way for researchers to build models
Machine learning, conversational dialog engine for creating chat bots
Powering Amazon custom machine learning chips
A ranked list of awesome machine learning Python libraries