LICENSE
README.md
pyproject.toml
setup.py
pymbbo/__init__.py
pymbbo/config.py
pymbbo.egg-info/PKG-INFO
pymbbo.egg-info/SOURCES.txt
pymbbo.egg-info/dependency_links.txt
pymbbo.egg-info/requires.txt
pymbbo.egg-info/top_level.txt
pymbbo/architectures/__init__.py
pymbbo/architectures/base_arch.py
pymbbo/architectures/asa_transformer/__init__.py
pymbbo/architectures/asa_transformer/asa_attention.py
pymbbo/architectures/asa_transformer/asa_block.py
pymbbo/architectures/asa_transformer/model.py
pymbbo/architectures/asa_transformer/triton_kernel.py
pymbbo/architectures/causal_matrix_merge/__init__.py
pymbbo/architectures/causal_matrix_merge/config.py
pymbbo/architectures/causal_matrix_merge/generation.py
pymbbo/architectures/causal_matrix_merge/merge.py
pymbbo/architectures/causal_matrix_merge/model.py
pymbbo/architectures/causal_matrix_merge/state.py
pymbbo/architectures/causal_matrix_merge_v2/__init__.py
pymbbo/architectures/causal_matrix_merge_v2/checkpoint_reader.py
pymbbo/architectures/causal_matrix_merge_v2/config.py
pymbbo/architectures/causal_matrix_merge_v2/layer.py
pymbbo/architectures/causal_matrix_merge_v2/merge_v2.py
pymbbo/architectures/causal_matrix_merge_v2/mlp.py
pymbbo/architectures/causal_matrix_merge_v2/model.py
pymbbo/architectures/cnn/__init__.py
pymbbo/architectures/cnn/model.py
pymbbo/architectures/mlp/__init__.py
pymbbo/architectures/mlp/model.py
pymbbo/architectures/transformer/__init__.py
pymbbo/architectures/transformer/model.py
pymbbo/data/__init__.py
pymbbo/data/batcher.py
pymbbo/data/dataset.py
pymbbo/engine/__init__.py
pymbbo/engine/callbacks.py
pymbbo/engine/trainer.py
pymbbo/metrics/__init__.py
pymbbo/metrics/benchmark.py
pymbbo/metrics/standard.py
pymbbo/models/__init__.py
pymbbo/models/base.py
pymbbo/models/factory.py
pymbbo/models/layers.py
pymbbo/models/persistence.py
pymbbo/models/registry.py
tests/test_asa_transformer.py
tests/test_integration_causal_matrix_merge.py
tests/test_integration_causal_matrix_merge_v2.py
tests/test_integration_forward_shape_verification.py
tests/test_pbt_causal_matrix_merge_v2.py
tests/test_pbt_slot_local_merge.py
tests/test_pymbbo.py