LICENSE
MANIFEST.in
README.md
pyproject.toml
assets/MLX_Kourkoutas.png
assets/t_2dframes.png
src/kbeta_transformer2d/__init__.py
src/kbeta_transformer2d/data.py
src/kbeta_transformer2d/demo_heat2d.py
src/kbeta_transformer2d/io_utils.py
src/kbeta_transformer2d/model.py
src/kbeta_transformer2d/optim_factory.py
src/kbeta_transformer2d/plot_utils.py
src/kbeta_transformer2d/py.typed
src/kbeta_transformer2d/train.py
src/kbeta_transformer2d/utils.py
src/kbeta_transformer2d.egg-info/PKG-INFO
src/kbeta_transformer2d.egg-info/SOURCES.txt
src/kbeta_transformer2d.egg-info/dependency_links.txt
src/kbeta_transformer2d.egg-info/requires.txt
src/kbeta_transformer2d.egg-info/top_level.txt
src/kbeta_transformer2d/configs/heat2d.yml
src/kbeta_transformer2d/configs/paper.yml
src/kbeta_transformer2d/configs/quick_test.yml
tests/test_imports.py
tests/test_smoke.py