Autonomous coding agent right in your IDE, capable of creating/editing files, executing commands, using the browser, and more with your permission every step of the way.

Official repo for paper "MiraData: A Large-Scale Video Dataset with Long Durations and Structured Captions"

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.

A high-throughput and memory-efficient inference and serving engine for LLMs

The standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

Tabular Deep Learning Library for PyTorch

MLX: An array framework for Apple silicon

Zipline, a Pythonic Algorithmic Trading Library

🛰️ An approximate nearest-neighbor search library for Python and Java with a focus on ease of use, simplicity, and deployability.

🌱 Substrate is a modern Copier template for scaffolding Python packages and apps

【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.

Graph Neural Network Library for PyTorch

A curated list of awesome things related to Pydantic! 🌪️

Convert Machine Learning Code Between Frameworks

This repo provides a customizable stack for starting new ML projects on Databricks that follow production best-practices out of the box.

Cookiecutter template for FastAPI projects using: Machine Learning, uv, Github Actions and Pytests

A tool for exploring each layer in a docker image

Source code for Twitter's Recommendation Algorithm

Source code for Twitter's Recommendation Algorithm

An extremely fast Python linter and code formatter, written in Rust.

Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk

Always know what to expect from your data.

A logical, reasonably standardized, but flexible project structure for doing and sharing data science work.

A playbook for systematically maximizing the performance of deep learning models.

A latent text-to-image diffusion model

Learn how to design systems at scale and prepare for system design interviews

FinRL®: Financial Reinforcement Learning. 🔥

A python package for interacting with Subaru STARLINK remote vehicle services.

Machine learning metrics for distributed, scalable PyTorch applications.