Optax is a gradient processing and optimization library for JAX
Sample code and notebooks for Generative AI on Google Cloud
A library for accelerating Transformer models on NVIDIA GPUs
Offline Text To Speech synthesis for python
HunyuanVideo: A Systematic Framework For Large Video Generation Model
A Python package to assess and improve fairness of ML models
A refreshing functional take on deep learning
Training PyTorch models with differential privacy
Fast forecasting with statistical and econometric models
ktrain is a Python library that makes deep learning AI more accessible
Library for serving Transformers models on Amazon SageMaker
A fast library for AutoML and tuning
A simple, secure MCP-to-OpenAPI proxy server
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Deep learning optimization library making distributed training easy
State-of-the-art diffusion models for image and audio generation
MiniSom is a minimalistic implementation of the Self Organizing Maps
Explainability and Interpretability to Develop Reliable ML models
Audiocraft is a library for audio processing and generation
Automatically Visualize any dataset, any size
Agents write python code to call tools and orchestrate other agents
A unified framework for scalable computing
Open-source AI research assistant for biomedicine
Tools for merging pretrained large language models
Library for OCR-related tasks powered by Deep Learning