.gitignore
LICENSE
README.md
pyproject.toml
.github/workflows/ci.yml
.github/workflows/publish.yml
eae_runtime/__init__.py
eae_runtime/_version.py
eae_runtime/adjoint.py
eae_runtime/backend.py
eae_runtime/blocks.py
eae_runtime/boundary_store.py
eae_runtime/config.py
eae_runtime/events.py
eae_runtime/forward_executor.py
eae_runtime/logging_utils.py
eae_runtime/memory.py
eae_runtime/pipeline.py
eae_runtime/profiler.py
eae_runtime/py.typed
eae_runtime/reconstruction.py
eae_runtime/runtime.py
eae_runtime.egg-info/PKG-INFO
eae_runtime.egg-info/SOURCES.txt
eae_runtime.egg-info/dependency_links.txt
eae_runtime.egg-info/requires.txt
eae_runtime.egg-info/scm_file_list.json
eae_runtime.egg-info/scm_version.json
eae_runtime.egg-info/top_level.txt
eae_runtime/contrib/__init__.py
eae_runtime/contrib/transformer_blocks.py
eae_runtime/passes/__init__.py
eae_runtime/passes/base.py
eae_runtime/passes/clip.py
eae_runtime/passes/logging_pass.py
eae_runtime/passes/quantize.py
eae_runtime/passes/regularization.py
eae_runtime/passes/synthetic_gradient.py
eae_runtime/schedulers/__init__.py
eae_runtime/schedulers/async_scheduler.py
eae_runtime/schedulers/base.py
eae_runtime/schedulers/distributed.py
eae_runtime/schedulers/pipeline_scheduler.py
eae_runtime/schedulers/sequential.py
examples/train_transformer_lm.py
tests/conftest.py
tests/test_adjoint.py
tests/test_backend.py
tests/test_boundary_store.py
tests/test_config_and_blocks.py
tests/test_contrib_transformer_blocks.py
tests/test_distributed_multiprocess.py
tests/test_end_to_end_transformer.py
tests/test_events.py
tests/test_gradient_equivalence.py
tests/test_memory_leaks.py
tests/test_memory_manager.py
tests/test_mixed_precision.py
tests/test_passes.py
tests/test_profiler.py
tests/test_reconstruction.py
tests/test_runtime.py
tests/test_schedulers.py