/spin-model-transformers

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

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers