A python library for self-supervised learning on images
Machine Learning automation and tracking
Machine learning on FPGAs using HLS
A TensorFlow library for applied reinforcement learning
JAX-based neural network library
JAX-based neural network library
Build portable, production-ready MLOps pipelines
TFX is an end-to-end platform for deploying production ML pipelines
SkyPilot: Run AI and batch jobs on any infra
AI agents autonomously run and improve ML experiments overnight
A unified, comprehensive and efficient recommendation library
A comprehensive quantitative trading system with AI-powered analysis
The most intuitive, flexible, way for researchers to build models
Clean, Robust, and Unified PyTorch implementation
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Spatiotemporal Signal Processing with Neural Machine Learning Models
Sandbox for training deep learning networks
Unified Model Serving Framework
Modular quant framework
Scalable machine learning for time series forecasting
Fault-tolerant, highly scalable GPU orchestration
Decomposable Multiscale Mixing for Time Series Forecasting
High-Performance Face Recognition Library on PaddlePaddle & PyTorch
Faster and easier training and deployments
Running large language models on a single GPU