LICENSE
NOTICE
README.md
pyproject.toml
composelm/__init__.py
composelm/build.py
composelm/config.py
composelm/py.typed
composelm/types.py
composelm.egg-info/PKG-INFO
composelm.egg-info/SOURCES.txt
composelm.egg-info/dependency_links.txt
composelm.egg-info/entry_points.txt
composelm.egg-info/requires.txt
composelm.egg-info/top_level.txt
composelm/bench/__init__.py
composelm/bench/metrics.py
composelm/bench/runner.py
composelm/convert/__init__.py
composelm/convert/hf.py
composelm/convert/vllm.py
composelm/distributed/__init__.py
composelm/distributed/deepspeed_utils.py
composelm/distributed/ep.py
composelm/distributed/fsdp.py
composelm/distributed/launch.py
composelm/distributed/strategy.py
composelm/infer/__init__.py
composelm/infer/batching.py
composelm/infer/generate.py
composelm/infer/speculative.py
composelm/infer/vllm.py
composelm/kernels/__init__.py
composelm/kernels/compile.py
composelm/kernels/flash.py
composelm/kernels/fused.py
composelm/layers/__init__.py
composelm/layers/attention.py
composelm/layers/embedding.py
composelm/layers/ffn.py
composelm/layers/linear.py
composelm/layers/mla.py
composelm/layers/moe.py
composelm/layers/norm.py
composelm/layers/pos_emb.py
composelm/layers/residual.py
composelm/models/__init__.py
composelm/models/block.py
composelm/models/registry.py
composelm/models/transformer.py
composelm/precision/__init__.py
composelm/precision/autocast.py
composelm/precision/policy.py
composelm/train/__init__.py
composelm/train/api.py
composelm/train/callbacks.py
composelm/train/checkpoint.py
composelm/train/optim.py
composelm/train/trainer.py
tests/test_activation_checkpointing.py
tests/test_all_arch_smoke.py
tests/test_attention.py
tests/test_batching.py
tests/test_bench.py
tests/test_benchmark_regression.py
tests/test_block_forward.py
tests/test_build.py
tests/test_checkpoint.py
tests/test_chunked_loss.py
tests/test_config.py
tests/test_convert_hf.py
tests/test_distributed_hooks.py
tests/test_distributed_weight_export.py
tests/test_embedding.py
tests/test_examples.py
tests/test_ffn.py
tests/test_generate.py
tests/test_kernels.py
tests/test_mla.py
tests/test_moe.py
tests/test_norm.py
tests/test_pos_emb.py
tests/test_precision.py
tests/test_registry.py
tests/test_release_metadata.py
tests/test_soak_checker.py
tests/test_speculative.py
tests/test_tracking_callbacks.py
tests/test_trainer_edges.py
tests/test_trainer_smoke.py
tests/test_training_runtime.py
tests/test_vllm.py
tests/test_yarn.py