timoklein

backpack

BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.

FriendsDontLetFriends

Friends don't let friends make certain types of data visualization - What are they and why are they bad.

flashbax

⚡ Flashbax: Accelerated Replay Buffers in JAX

TACO

Code for "TACO: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement Learning"

d2l-en

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

xland-minigrid

JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️

stat_rethinking_2024

haliax

Named Tensors for Legible Deep Learning in JAX

numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

chex

sbx

SBX: Stable Baselines Jax (SB3 + Jax)

alphazero-gym

AlphaZero for continuous control tasks

mctx

Monte Carlo tree search in JAX

nvitop

An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.

gymnax-blines

Baselines for gymnax 🤖

navix

Accelerated minigrid environments with JAX

helx

Interoperating between (Deep) Reiforcement Learning libraries

mlxtend

A library of extension and helper modules for Python's data analysis and machine learning libraries.

ppo_jax

Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights on all environments.

jaxrl2

jaxrl

JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

gymnax

RL Environments in JAX 🌍

purejaxrl

Really Fast End-to-End Jax RL Implementations

RLBench

A large-scale benchmark and learning environment.

mermaid

Generation of diagrams like flowcharts or sequence diagrams from text in a similar manner as markdown

tyro

Zero-effort CLI interfaces & config objects, from types

pandoc-beamer-how-to

pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

HowToLiveLonger

程序员延寿指南 | A programmer's guide to live longer

torchmetrics

Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.