An Easy-to-use, Scalable and High-performance RLHF Framework
On-device AI across mobile, embedded and edge for PyTorch
A library for audio and music analysis, feature extraction
A fast library for AutoML and tuning
NVIDIA Federated Learning Application Runtime Environment
A lightweight vision library for performing large object detection
Physical Symbolic Optimization
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Machine learning metrics for distributed, scalable PyTorch application
AIMET is a library that provides advanced quantization and compression
A fast image processing library with low memory needs
OpenMLDB is an open-source machine learning database
Optax is a gradient processing and optimization library for JAX
Python toolbox to create adversarial examples
A simple but complete full-attention transformer
A library of extension and helper modules for Python's data analysis
A Python Package to Tackle the Curse of Imbalanced Datasets in ML
Magnetoencephalography (MEG) and Electroencephalography EEG in Python
An MLOps framework to package, deploy, monitor and manage models
Python Stream Processing
Algorithmic Trading in Python with Machine Learning
Analyzing, storing and visualizing big data, scientifically
Python Package for ML-Based Heterogeneous Treatment Effects Estimation
Deep learning optimization library making distributed training easy