Open-source Python framework for training and backtesting financial trading agents with deep reinforcement learning.
Open-source C++ library of computer vision and image processing algorithms, with bindings for Python, Java, and other languages.
PyTorch library for audio generation research, providing inference and training code for MusicGen, AudioGen, and EnCodec.
A library of model optimization techniques (quantization, pruning, distillation, NAS, speculative decoding, sparsity) that exports compressed checkpoints for inference frameworks.
An end-to-end open source platform for building and deploying machine learning models, with Python and C++ APIs.
Python library providing GPU-accelerated tensor computation like NumPy, plus a tape-based autograd system for building deep neural networks.