JAX-based neural network library
A refreshing functional take on deep learning
AI Toolkit for Healthcare Imaging
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, etc
An Easy-to-use, Scalable and High-performance RLHF Framework
Reference implementations of MLPerf™ training benchmarks
PyTorch version of Stable Baselines
An easy-to-use & supercharged open-source experiment tracker
A simple but complete full-attention transformer
A game theoretic approach to explain the output of ml models
JAX-based neural network library
Implementation of Imagen, Google's Text-to-Image Neural Network
Open-Source AI Camera. Empower any camera/CCTV
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
PyTorch extensions for fast R&D prototyping and Kaggle farming
Statistical library designed to fill the void in Python's time series
A library for accelerating Transformer models on NVIDIA GPUs
AutoGluon: AutoML for Image, Text, and Tabular Data
A PyTorch-based Speech Toolkit
The most intuitive, flexible, way for researchers to build models
Open Source Computer Vision Library
Superfast AI decision making and processing of multi-modal data
A system for quickly generating training data with weak supervision
Open source framework for deep learning satellite and aerial imagery
Fast image augmentation library and an easy-to-use wrapper