Lightning fast C++/CUDA neural network framework
Python-based neural networks API
Create videos with Stable Diffusion
A python library for self-supervised learning on images
Probabilistic time series modeling in Python
A python library for easy manipulation and forecasting of time series
DeepMind's software stack for physics-based simulation
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, etc
Solve end to end problems using Llama model family
Optax is a gradient processing and optimization library for JAX
A very simple framework for state-of-the-art NLP
TensorFlow-based neural network library
A system for quickly generating training data with weak supervision
The PyTorch-based audio source separation toolkit for researchers
A lightweight vision library for performing large object detection
Graph Neural Network Library for PyTorch
The Operator Splitting QP Solver
Deep Learning Visualization Toolkit
Tensor Learning in Python
A unified interface for distributed computing
Python toolbox to create adversarial examples
Meta-Transformer for Unified Multimodal Learning
Real-Time High-Resolution Background Matting
Reference implementations of MLPerf™ training benchmarks
TFX is an end-to-end platform for deploying production ML pipelines