LICENSE
MANIFEST.in
README.md
pyproject.toml
parallax/__init__.py
parallax/cli.py
parallax/config.py
parallax/benchmark/__init__.py
parallax/benchmark/comparator.py
parallax/benchmark/reporting.py
parallax/benchmark/runner.py
parallax/configs/maxtext_bf16.yml
parallax/configs/maxtext_fp8.yml
parallax/configs/maxtext_llama100m_base.yml
parallax/configs/maxtext_nvfp4.yml
parallax/data/__init__.py
parallax/data/tinystories.py
parallax/engines/__init__.py
parallax/engines/base.py
parallax/engines/maxtext_engine.py
parallax/engines/precision_manager.py
parallax/engines/torch_engine.py
parallax/models/__init__.py
parallax/models/llama.py
parallax/models/rope.py
parallax/utils/__init__.py
parallax/utils/flops.py
parallax/utils/hardware.py
scripts/run_benchmarks.sh
scripts/setup_remote_blackwell.sh
slm_parallax.egg-info/PKG-INFO
slm_parallax.egg-info/SOURCES.txt
slm_parallax.egg-info/dependency_links.txt
slm_parallax.egg-info/entry_points.txt
slm_parallax.egg-info/requires.txt
slm_parallax.egg-info/top_level.txt
tests/test_cli.py
tests/test_data.py
tests/test_model.py
tests/test_precision.py