mcbal

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

equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

fast-disentangle-pytorch

julia

The Julia Language: A fresh approach to technical computing.

mcbal