The CUDA target for Numba
A NumPy-compatible array library accelerated by CUDA
Performance meets Productivity
A GUI tool for extracting hard-coded subtitle (hardsub) from videos
The most powerful and modular diffusion model GUI, api and backend
ProtoMotions is a GPU-accelerated simulation and learning framework
Development repository for the Triton language and compiler
Deep learning optimization library making distributed training easy
An open source library for GPU-accelerated robot learning
DeepEP: an efficient expert-parallel communication library
A Python framework for accelerated simulation, data generation
Meridian is an MMM framework
Trainable latent-memory framework for 100M-token contexts
Standardized Serverless ML Inference Platform on Kubernetes
Open Source Differentiable Computer Vision Library
A bidirectional pipeline parallelism algorithm
Library of JAX-based building blocks for reinforcement learning agents
2D and 3D Face alignment library build using pytorch
FAIR Chemistry's library of machine learning methods for chemistry
A set of Docker images for training and serving models in TensorFlow
Library for OCR-related tasks powered by Deep Learning
A cross-platform Python library for differentiable programming
Enables the best performance on NVIDIA RTX Graphics Cards
Public CI, Docker images for popular JAX libraries
NumPy aware dynamic Python compiler using LLVM