Matthias Bal

deep-implicit-attention. Implementation of deep implicit attention in PyTorch

spin-model-transformers. Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX

afem. Implementation of approximate free-energy minimization in PyTorch

mcbal.github.io. HTML