LICENSE
README.md
pyproject.toml
foundax/__init__.py
foundax/_callable_module.py
foundax/_vendors.py
foundax/bcat.py
foundax/cli.py
foundax/combinators.py
foundax/dpot.py
foundax/morph.py
foundax/mpp.py
foundax/nn.py
foundax/pdeformer2.py
foundax/pipe.py
foundax/poseidon.py
foundax/prose.py
foundax/walrus.py
foundax.egg-info/PKG-INFO
foundax.egg-info/SOURCES.txt
foundax.egg-info/dependency_links.txt
foundax.egg-info/entry_points.txt
foundax.egg-info/requires.txt
foundax.egg-info/top_level.txt
foundax/architectures/__init__.py
foundax/architectures/cno.py
foundax/architectures/common.py
foundax/architectures/deeponet.py
foundax/architectures/fno.py
foundax/architectures/geofno.py
foundax/architectures/gnot.py
foundax/architectures/linear.py
foundax/architectures/mgno.py
foundax/architectures/mlp.py
foundax/architectures/pcno.py
foundax/architectures/pit.py
foundax/architectures/pointnet.py
foundax/architectures/transformer.py
foundax/architectures/unet.py
foundax/layers/__init__.py
foundax/layers/blocks.py
repos/jax_bcat/jax_bcat/__init__.py
repos/jax_bcat/jax_bcat/configs.py
repos/jax_bcat/jax_bcat/model_eqx.py
repos/jax_dpot/jax_dpot/__init__.py
repos/jax_dpot/jax_dpot/configs.py
repos/jax_dpot/jax_dpot/model_eqx.py
repos/jax_dpot/jax_dpot/utils.py
repos/jax_morph/jax_morph/__init__.py
repos/jax_morph/jax_morph/configs.py
repos/jax_morph/jax_morph/model_eqx.py
repos/jax_morph/jax_morph/patchify.py
repos/jax_morph/jax_morph/positional_encoding.py
repos/jax_mpp/jax_mpp/__init__.py
repos/jax_mpp/jax_mpp/avit_eqx.py
repos/jax_mpp/jax_mpp/configs.py
repos/jax_mpp/jax_mpp/convert_weights.py
repos/jax_pdeformer2/jax_pdeformer2/__init__.py
repos/jax_pdeformer2/jax_pdeformer2/convert_checkpoint.py
repos/jax_pdeformer2/jax_pdeformer2/model_eqx.py
repos/jax_poseidon/jax_poseidon/__init__.py
repos/jax_poseidon/jax_poseidon/scot_eqx.py
repos/jax_prose/jax_prose/__init__.py
repos/jax_prose/jax_prose/config.py
repos/jax_prose/jax_prose/model_eqx.py
repos/jax_walrus/jax_walrus/__init__.py
repos/jax_walrus/jax_walrus/convert_weights.py
repos/jax_walrus/jax_walrus/model_eqx.py
tests/test_bcat_eqx.py
tests/test_dispatch.py
tests/test_dpot_eqx.py
tests/test_heavy_models.py
tests/test_morph_eqx.py
tests/test_mpp_eqx.py
tests/test_namespace.py
tests/test_pdeformer2_eqx.py
tests/test_pipe.py
tests/test_pipe_extended.py
tests/test_poseidon_eqx.py
tests/test_prose_eqx.py
tests/test_training.py
tests/test_vendors.py
tests/test_walrus_eqx.py