Build cross-modal and multimodal applications on the cloud
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
PyTorch code and models for V-JEPA self-supervised learning from video
An implementation of a deep learning recommendation model (DLRM)
Self-supervised visual learning using momentum contrast in PyTorch
Making large AI models cheaper, faster and more accessible
Graph Neural Network Library for PyTorch
PyTorch code and models for VJEPA2 self-supervised learning from video
Enabling PyTorch on Google TPU
Hub of ready-to-use datasets for ML models
Jupyter notebooks that demonstrate how to build models using SageMaker
Open source framework for deep learning satellite and aerial imagery
Deep Learning Visualization Toolkit
Simple and distributed Machine Learning
A GPU-accelerated library containing highly optimized building blocks
Open Source Computer Vision Library
An industrial grade federated learning framework
ktrain is a Python library that makes deep learning AI more accessible
Deep universal probabilistic programming with Python and PyTorch
Audiocraft is a library for audio processing and generation
Implementation of Make-A-Video, new SOTA text to video generator
Easy-to-use deep learning framework with 3 key features
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, etc
OneFlow is a deep learning framework designed to be user-friendly
OpenMMLab Model Deployment Framework