A TensorFlow library for applied reinforcement learning
Unified Model Serving Framework
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
PyTorch extensions for fast R&D prototyping and Kaggle farming
TFX is an end-to-end platform for deploying production ML pipelines
Pretrained (Language) Models for Probabilistic Time Series Forecasting
SkyPilot: Run AI and batch jobs on any infra
Open-source tool designed to enhance the efficiency of workloads
Graph Neural Network Library for PyTorch
Models and examples built with TensorFlow
Open source framework for deep learning satellite and aerial imagery
Fast image augmentation library and an easy-to-use wrapper
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Powering Amazon custom machine learning chips
An unsupervised and free tool for image and video dataset analysis
Physical Symbolic Optimization
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
An MLOps framework to package, deploy, monitor and manage models
Create HTML profiling reports from pandas DataFrame objects
Clean, Robust, and Unified PyTorch implementation
Collection of useful data science topics along with articles
This project is a common knowledge point and code implementation
An agentic Machine Learning Engineer
Machine Learning Systems: Design and Implementation
Optax is a gradient processing and optimization library for JAX