JAX-based neural network library
TFX is an end-to-end platform for deploying production ML pipelines
SkyPilot: Run AI and batch jobs on any infra
Core ML tools contain supporting tools for Core ML model conversion
Deep and Machine Learning for Microscopy
A unified, comprehensive and efficient recommendation library
Trainable models and NN optimization tools
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
A comprehensive quantitative trading system with AI-powered analysis
Unified Model Serving Framework
Modular quant framework
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
Build portable, production-ready MLOps pipelines
AI agents autonomously run and improve ML experiments overnight
A lightweight vision library for performing large object detection
A refreshing functional take on deep learning
The most intuitive, flexible, way for researchers to build models
Build cross-modal and multimodal applications on the cloud
Scalable machine learning for time series forecasting
Powering Amazon custom machine learning chips