Extending JAX with custom C++ and CUDA code
? Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Deep Learning for humans
A high-throughput and memory-efficient inference and serving engine for LLMs
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
? Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
? A ranked list of awesome machine learning Python libraries. Updated weekly.
SGLang is a fast serving framework for large language models and vision language models.
kaldi-asr/kaldi is the official location of the Kaldi project.
Open3D: A Modern Library for 3D Data Processing
