Alexander Tschantz's repositories
pymdp_pddl
Generate active inference agents from PDDL
vl-tutorial
JAX implementation of "A primer on Variational Laplace"
alec-tschantz.github.io
Github Pages template for academic personal websites, forked from mmistakes/minimal-mistakes
dreamerv3-torch
Implementation of Dreamer v3 in pytorch.
dvi-jax
An implementation of Deterministic Variational Inference in jax
equinox-llama
Equinox implementation of llama3 and llama3.1
euclid
Geometry primitives (basic linear algebra) for Rust
fresh-chrome-extension
Useful bare minimum Google Chrome extension boilerplate.
gen-capsules
A package for modelling capsule networks in terms of generative models and inference
gpt_index
An index created by GPT to organize external information and answer queries!
iris
Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.
jpc
Flexible Inference for Predictive Coding Networks in JAX.
kalman_gmm
Kalman filters with Gaussian Mixture Model observations
koopman-jax
Consistent Koopman Autoencoders
nanoRLHF
RLHF experiments on a single A100 40G GPU. Support PPO, GRPO, REINFORCE, RAFT, RLOO, ReMax, DeepSeek R1-Zero reproducing.
tiny-grpo
Minimal hackable GRPO implementation
TinyZero
Clean, minimal, accessible reproduction of DeepSeek R1-Zero