Andrin Rehmann's repositories
typeshaders
WebGL shader playground in Typescript
jax-image-autoencoder
Quick dirty implementation of a convolutional image autoencoder with JAX, Flax and optax.
webgpu-particles
Minimal WebGPU Compute Shader Particle System
webgpu-noodles
High performance noodles using webgpu.
advanced-quantum-algorithms
Notes and exercise solutions for the course advanced quantum algorithms taught @UZH 2023.
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
jax-gan
Quick & dirty implementation of a generative adversarial network with jax, optax and equinox.
jax-neural-ode
Short implementation of the Neural ODE, including the adjoint based version in pure JAX and OPTAX
motion-derivative
Differentiable particle simulation for drawing images
pbdl-book
Welcome to the Physics-based Deep Learning Book (v0.2)
phishing
A curated list of known less-than-honest operators on Polkadot and Substrate networks. Includes a simple JS utility function to check any host or address against this list.
pigvae
Implementation of the Paper "Permutation-Invariant Variational Autoencoder for Graph-Level Representation Learning" by Robin Winter, Frank Noe and Djork-Arne Clevert.
pmwd
Differentiable Cosmological Forward Model
popjaxrl
Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
pytorch_geometric
Graph Neural Network Library for PyTorch
rl-fluids
Controling IC's for fluids using RL in JAX, Optax, Equinox and Phiflow.
SiGAT
source code for signed graph attention networks (ICANN2019) & SDGNN (AAAI2021)
sketches-nannou
Some creative coding sketches using nannou (rust)