LICENSE.txt
README.md
pyproject.toml
setup.cfg
tests/test_instantiate.py
tests/test_substack_trjs.py
trainax/__init__.py
trainax/_general_trainer.py
trainax/_mixer.py
trainax/_sample_data.py
trainax/_utils.py
trainax.egg-info/PKG-INFO
trainax.egg-info/SOURCES.txt
trainax.egg-info/dependency_links.txt
trainax.egg-info/requires.txt
trainax.egg-info/top_level.txt
trainax/callback/__init__.py
trainax/callback/_base.py
trainax/callback/_composite.py
trainax/callback/_get_network.py
trainax/callback/_grad_norm.py
trainax/callback/_loss.py
trainax/callback/_save_network.py
trainax/callback/_weight_norm.py
trainax/configuration/__init__.py
trainax/configuration/_base_configuration.py
trainax/configuration/_composite.py
trainax/configuration/_diverted_chain.py
trainax/configuration/_diverted_chain_branch_one.py
trainax/configuration/_mix_chain_post_physics.py
trainax/configuration/_residuum.py
trainax/configuration/_supervised.py
trainax/loss/__init__.py
trainax/loss/_base_loss.py
trainax/loss/_composite.py
trainax/loss/_lp_loss.py
trainax/loss/_mae_loss.py
trainax/loss/_mse_loss.py
trainax/loss/_rmse_loss.py
trainax/trainer/__init__.py
trainax/trainer/_diverted_chain_branch_one.py
trainax/trainer/_residuum.py
trainax/trainer/_supervised.py