yfw

Open-MAGVIT2

Open-MAGVIT2: Democratizing Autoregressive Visual Generation

gisting

Learning to Compress Prompts with Gist Tokens - https://arxiv.org/abs/2304.08467

wasm-postflop

[Development suspended] Advanced open-source Texas Hold'em GTO solver with optimized performance (web browser version)

fzf

:cherry_blossom: A command-line fuzzy finder

webdataset

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

tuning_playbook

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

sheetjs

📗 SheetJS Spreadsheet Data Toolkit -- New home https://git.sheetjs.com/SheetJS/sheetjs

miniwob-plusplus

MiniWoB++: a web interaction benchmark for reinforcement learning

project-layout

Standard Go Project Layout

tueplots

Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other scientific publications.

benchmark_VAE

Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)

rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

wrangl

Parallel data preprocessing for NLP and ML.

tianshou

An elegant PyTorch deep reinforcement learning library.

accelerate

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

lightning-hydra-template

Deep Learning project template best practices with Pytorch Lightning, Hydra, Tensorboard.

seed_rl

SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's architecture.

attrs

Python Classes Without Boilerplate

x-transformers

A simple but complete full-attention transformer with a set of promising experimental features from various papers

deepul

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

auriga

Auriga is a minimalist LaTeX beamer presentation theme 📽

micrograd

A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API

Minigrid

Simple and easily configurable grid world environments for reinforcement learning

torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

py-motmetrics

:bar_chart: Benchmark multiple object trackers (MOT) in Python

reloading

Change Python code while it's running without losing state

scalene

Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals

typer

Typer, build great CLIs. Easy to code. Based on Python type hints.

Miniworld

Simple and easily configurable 3D FPS-game-like environments for reinforcement learning