Optax is a gradient processing and optimization library for JAX
Machine learning metrics for distributed, scalable PyTorch application
Lazy Predict help build a lot of basic models without much code
An Easy-to-use, Scalable and High-performance RLHF Framework
Extensible, parallel implementations of t-SNE
A high performance implementation of HDBSCAN clustering
JAX-based neural network library
Book about interpretable machine learning
A fast library for AutoML and tuning
Jittor is a high-performance deep learning framework
DeepMind's software stack for physics-based simulation
A general fine-tuning kit geared toward image/video/audio diffusion
An open source implementation of CLIP
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
AI discovers 520000 stable inorganic crystal structures for research
Tool for visualizing and tracking your machine learning experiments
Open-source deep-learning framework for building and training
Core ML tools contain supporting tools for Core ML model conversion
Build MLOps Pipelines in Minutes
Deep and Machine Learning for Microscopy
A unified, comprehensive and efficient recommendation library
Statistical library designed to fill the void in Python's time series
A curated list of data mining papers about fraud detection
AIMET is a library that provides advanced quantization and compression
Machine learning image inpainting task that removes watermarks