LICENSE.txt
README.md
pyproject.toml
skalax/__init__.py
skalax/convert_weights.py
skalax/py.typed
skalax.egg-info/PKG-INFO
skalax.egg-info/SOURCES.txt
skalax.egg-info/dependency_links.txt
skalax.egg-info/requires.txt
skalax.egg-info/top_level.txt
skalax/functional/__init__.py
skalax/functional/base.py
skalax/functional/functions.py
skalax/functional/layers.py
skalax/functional/model.py
skalax/pyscf/__init__.py
skalax/pyscf/backend.py
skalax/pyscf/dft.py
skalax/pyscf/features.py
skalax/pyscf/numint.py
skalax/utils/__init__.py
skalax/utils/scatter.py
skalax/weights/config.json
skalax/weights/skala_buffers.npz
skalax/weights/skala_weights.npz
tests/test_full_model.py
tests/test_jax_equivalence.py
tests/test_jit.py
tests/test_nonlocal_model.py
tests/test_pyscf_integration.py
tests/test_with_loaded_weights.py