README.md
pyproject.toml
apax/__init__.py
apax/calibration.py
apax.egg-info/PKG-INFO
apax.egg-info/SOURCES.txt
apax.egg-info/dependency_links.txt
apax.egg-info/entry_points.txt
apax.egg-info/requires.txt
apax.egg-info/top_level.txt
apax/bal/__init__.py
apax/bal/api.py
apax/bal/feature_maps.py
apax/bal/kernel.py
apax/bal/selection.py
apax/bal/transforms.py
apax/cli/__init__.py
apax/cli/apax_app.py
apax/cli/templates/__init__.py
apax/config/__init__.py
apax/config/__init__.pyi
apax/config/common.py
apax/config/lr_config.py
apax/config/md_config.py
apax/config/model_config.py
apax/config/train_config.py
apax/data/__init__.py
apax/data/__init__.pyi
apax/data/initialization.py
apax/data/input_pipeline.py
apax/data/preprocessing.py
apax/data/statistics.py
apax/layers/__init__.py
apax/layers/activation.py
apax/layers/distances.py
apax/layers/empirical.py
apax/layers/initializers.py
apax/layers/masking.py
apax/layers/ntk_linear.py
apax/layers/properties.py
apax/layers/readout.py
apax/layers/scaling.py
apax/layers/descriptor/__init__.py
apax/layers/descriptor/basis_functions.py
apax/layers/descriptor/equiv_mp.py
apax/layers/descriptor/gaussian_moment_descriptor.py
apax/layers/descriptor/moments.py
apax/layers/descriptor/so3krates.py
apax/layers/descriptor/triangular_indices.py
apax/md/__init__.py
apax/md/__init__.pyi
apax/md/ase_calc.py
apax/md/constraints.py
apax/md/dynamics_checks.py
apax/md/function_transformations.py
apax/md/io.py
apax/md/md_checkpoint.py
apax/md/schedules.py
apax/md/sim_utils.py
apax/md/simulate.py
apax/nn/__init__.py
apax/nn/__init__.pyi
apax/nn/builder.py
apax/nn/models.py
apax/nodes/__init__.py
apax/nodes/analysis.py
apax/nodes/md.py
apax/nodes/model.py
apax/nodes/selection.py
apax/nodes/utils.py
apax/optimizer/__init__.py
apax/optimizer/get_optimizer.py
apax/optimizer/optimizers.py
apax/train/__init__.py
apax/train/__init__.pyi
apax/train/callbacks.py
apax/train/checkpoints.py
apax/train/eval.py
apax/train/loss.py
apax/train/metrics.py
apax/train/mlflow.py
apax/train/parameters.py
apax/train/run.py
apax/train/trainer.py
apax/transfer_learning/__init__.py
apax/transfer_learning/parameter_transfer.py
apax/utils/__init__.py
apax/utils/__init__.pyi
apax/utils/convert.py
apax/utils/data.py
apax/utils/datasets.py
apax/utils/helpers.py
apax/utils/math.py
apax/utils/random.py
apax/utils/transform.py
apax/utils/jax_md_reduced/__init__.py
apax/utils/jax_md_reduced/dataclasses.py
apax/utils/jax_md_reduced/partition.py
apax/utils/jax_md_reduced/quantity.py
apax/utils/jax_md_reduced/simulate.py
apax/utils/jax_md_reduced/smap.py
apax/utils/jax_md_reduced/space.py
apax/utils/jax_md_reduced/util.py