Andrin Rehmann's repositories
rl-fluids
Controling IC's for fluids using RL in JAX, Optax, Equinox and Phiflow.
jax-gan
Quick & dirty implementation of a generative adversarial network with jax, optax and equinox.
jax-image-autoencoder
Quick dirty implementation of a convolutional image autoencoder with JAX, Flax and optax.
SiGAT
source code for signed graph attention networks (ICANN2019) & SDGNN (AAAI2021)
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
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)]
advanced-quantum-algorithms
Notes and exercise solutions for the course advanced quantum algorithms taught @UZH 2023.
webgpu-noodles
High performance noodles using webgpu.
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
webgpu-particles
Minimal WebGPU Compute Shader Particle System
sketches-nannou
Some creative coding sketches using nannou (rust)
pbdl
Exploration of Differentiable Physics aka the simulation derivative
asim
Advanced Simulation in Natural Science
pbdl-book
Welcome to the Physics-based Deep Learning Book (v0.2)
pytorch_geometric
Graph Neural Network Library for PyTorch
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.
typeshaders
WebGL shader playground in Typescript
pigvae
Implementation of the Paper "Permutation-Invariant Variational Autoencoder for Graph-Level Representation Learning" by Robin Winter, Frank Noe and Djork-Arne Clevert.