SkyPilot: Run AI and batch jobs on any infra
AutoML library for deep learning
Modular quant framework
Graph Neural Network Library for PyTorch
Tensor Learning in Python
Models for the spaCy Natural Language Processing (NLP) library
Library to help with training and evaluating neural networks
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Multi-tool for semantic search
Helps data scientists define testable self-documenting dataflows
Geometric deep learning extension library for PyTorch
AI agents autonomously run and improve ML experiments overnight
A PyTorch-based Speech Toolkit
Data science on data without acquiring a copy
Data loaders and abstractions for text and NLP
A system for quickly generating training data with weak supervision
The Pocket Datalab
The PyTorch-based audio source separation toolkit for researchers
Graph Neural Networks with Keras and Tensorflow 2
Official Python Implementation for "3D Multi-Object Tracking
Fast Forward Computer Vision (and other ML workloads!)
pyntcloud is a Python library for working with 3D point clouds
gradslam is an open source differentiable dense SLAM library
Convolutional Neural Networks to predict aesthetic quality of images
World's simplest facial recognition api for Python & the command line