Stars
A bidirectional pipeline parallelism algorithm for computation-communication overlap in V3/R1 training.
A Data Streaming Library for Efficient Neural Network Training
Prodigy and Schedule-Free, together at last.
🔥 Blazing fast bulk data transfers between any cloud 🔥
👁️ 🖼️ 🔥PyTorch Toolbox for Image Quality Assessment, including PSNR, SSIM, LPIPS, FID, NIQE, NRQM(Ma), MUSIQ, TOPIQ, NIMA, DBCNN, BRISQUE, PI and more...
The open-source visual AI programming environment and TypeScript library
A library for unit scaling in PyTorch
Task-Aware Agent-driven Prompt Optimization Framework
Moxin is a family of fully open-source and reproducible LLMs
streamline the fine-tuning process for multimodal models: PaliGemma 2, Florence-2, and Qwen2.5-VL
A suite of image and video neural tokenizers
[MLSys 2024 Best Paper Award] AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration
[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning
Quantized Attention achieves speedup of 2-5x and 3-11x compared to FlashAttention and xformers, without lossing end-to-end metrics across language, image, and video models.
A more helpful way to view differences between complex data structures in RSpec.
Causal Discovery in Python. It also includes (conditional) independence tests and score functions.
Efficient Learning of Augmentation Policy Schedules
🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools
JuiceFS is a distributed POSIX file system built on top of Redis and S3.
IVON optimizer for neural networks based on variational learning.
A tool to generate a daily digest of your Gmail inbox. Helpful if you have agents working on your behalf.
A python library for user-friendly forecasting and anomaly detection on time series.
A fast and memory efficient image library in Rust