LICENSE
README.md
pyproject.toml
src/mentat_lss/__init__.py
src/mentat_lss/cosmo_utils.py
src/mentat_lss/dataset.py
src/mentat_lss/emulator.py
src/mentat_lss/gpsclass.py
src/mentat_lss/training_loops.py
src/mentat_lss/utils.py
src/mentat_lss.egg-info/PKG-INFO
src/mentat_lss.egg-info/SOURCES.txt
src/mentat_lss.egg-info/dependency_links.txt
src/mentat_lss.egg-info/requires.txt
src/mentat_lss.egg-info/top_level.txt
src/mentat_lss/_vendor/__init__.py
src/mentat_lss/_vendor/fastpt/FASTPT.py
src/mentat_lss/_vendor/fastpt/FASTPT_original.py
src/mentat_lss/_vendor/fastpt/FASTPT_simple.py
src/mentat_lss/_vendor/fastpt/HT.py
src/mentat_lss/_vendor/fastpt/IA.py
src/mentat_lss/_vendor/fastpt/IA_ABD.py
src/mentat_lss/_vendor/fastpt/IA_ta.py
src/mentat_lss/_vendor/fastpt/IA_tt.py
src/mentat_lss/_vendor/fastpt/J_k.py
src/mentat_lss/_vendor/fastpt/J_table.py
src/mentat_lss/_vendor/fastpt/OV.py
src/mentat_lss/_vendor/fastpt/P_extend.py
src/mentat_lss/_vendor/fastpt/RG_RK4.py
src/mentat_lss/_vendor/fastpt/RG_RK4_filt.py
src/mentat_lss/_vendor/fastpt/RG_STS.py
src/mentat_lss/_vendor/fastpt/RG_ani.py
src/mentat_lss/_vendor/fastpt/RSD.py
src/mentat_lss/_vendor/fastpt/RSD_ItypeII.py
src/mentat_lss/_vendor/fastpt/Wigner_symbols.py
src/mentat_lss/_vendor/fastpt/__init__.py
src/mentat_lss/_vendor/fastpt/bias_choice_plot.py
src/mentat_lss/_vendor/fastpt/fastpt_extr.py
src/mentat_lss/_vendor/fastpt/filter_Pk.py
src/mentat_lss/_vendor/fastpt/gamma_funcs.py
src/mentat_lss/_vendor/fastpt/get_nu1_nu2.py
src/mentat_lss/_vendor/fastpt/info.py
src/mentat_lss/_vendor/fastpt/initialize_params.py
src/mentat_lss/_vendor/fastpt/kPol.py
src/mentat_lss/_vendor/fastpt/matter_power_spt.py
src/mentat_lss/_vendor/fastpt/matter_power_spt_updated_G3_factor.py
src/mentat_lss/_vendor/fastpt/param_matrix.py
src/mentat_lss/_vendor/symbolic_pofk/__init__.py
src/mentat_lss/_vendor/symbolic_pofk/linear.py
src/mentat_lss/models/analytic_terms.py
src/mentat_lss/models/blocks.py
src/mentat_lss/models/single_transformer.py
src/mentat_lss/models/stacked_mlp.py
src/mentat_lss/models/stacked_transformer.py
tests/test_analytic_terms.py
tests/test_network.py
tests/test_utils.py