SkyPilot: Run AI and batch jobs on any infra
Multi-tool for semantic search
Modular quant framework
AutoML library for deep learning
Graph Neural Network Library for PyTorch
Tensor Learning in Python
Models for the spaCy Natural Language Processing (NLP) library
A PyTorch-based Speech Toolkit
Library to help with training and evaluating neural networks
Geometric deep learning extension library for PyTorch
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Helps data scientists define testable self-documenting dataflows
AI agents autonomously run and improve ML experiments overnight
Data science on data without acquiring a copy
Offline speech recognition API for Android, iOS, Raspberry Pi
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!)
Providing the solutions for high-frequency trading (HFT) strategies
pyntcloud is a Python library for working with 3D point clouds