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
Multi-tool for semantic search
This repository is a curated collection of links to various courses
DeepVariant is an analysis pipeline that uses a deep neural networks
Open source framework for deep learning satellite and aerial imagery
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
AI agents autonomously run and improve ML experiments overnight
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, etc
Clean, Robust, and Unified PyTorch implementation
Collection of useful data science topics along with articles
A Collection of Cheatsheets, Books, Questions, and Portfolio
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
Sacred is a tool to help you configure, andorganize IDSIA experiments