RobertTLange

ml-engineering

Machine Learning Engineering Open Book

pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques

PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.

awesome-jax

JAX - A curated list of resources https://github.com/google/jax

brain-tokyo-workshop

🧠🗼

envpool

C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.

learned_optimization

caliban

Research workflows made easy, locally and in the Cloud.

tpu-starter

Everything you want to know about Google Cloud TPU

mle-monitor

Lightweight Experiment & Resource Monitoring 📺

pypop

PyPop7: A Pure-Python Library for POPulation-based Black-Box Optimization (BBO), especially their *Large-Scale* versions/variants. https://pypop.rtfd.io/

mle-hyperopt

Lightweight Hyperparameter Optimization 🚂

hypnettorch

Package for working with hypernetworks in PyTorch.

jax-resnet

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

hyper-nn

Easy Hypernetworks in Pytorch and Jax

rljax

A collection of RL algorithms written in JAX.

atari-agents

Code and links for over 25,000 trained Atari agents

Python_Benchmark_Test_Optimization_Function_Single_Objective

stui

A Slurm dashboard for the terminal.

DeepHyperNEAT

A public python implementation of the DeepHyperNEAT system for evolving neural networks. Developed by Felix Sosa and Kenneth Stanley. See paper here: https://eplex.cs.ucf.edu/papers/sosa_ugrad_report18.pdf

popjaxrl

Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]

multi-rl-crowd-sim

Multi-Agent Reinforcement Learning of Crowd Simulation in GPU (PyTorch)

ppo_jax

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

mle-scheduler

Lightweight Cluster/Cloud VM Job Management 🚀

HPO-B

[NeurIPS DBT 2021] HPO-B

minGPT-flax

GPT implementation in Flax

metax

flexible meta-learning in jax

pureRL

shrink-perturb

Optax implementation of shrink and perturb (Ash & Adams, 2020).

PygameModelling22

This is the offcicial repo of the pygame modelling course for collective systems workshop 2022 Berlin