LICENSE
README.md
pyproject.toml
GenZ/__init__.py
GenZ/analyse_model.py
GenZ/collective_times.py
GenZ/operator_base.py
GenZ/operators.py
GenZ/parallelism.py
GenZ/system.py
GenZ/unit.py
GenZ/version.py
GenZ/Astra_sim/fix_chakra_traces.py
GenZ/Astra_sim/get_astra_sim_time.py
GenZ/LLM_inference/__init__.py
GenZ/LLM_inference/best_parallelization.py
GenZ/LLM_inference/llm_decode.py
GenZ/LLM_inference/llm_prefill.py
GenZ/LLM_inference/platform_size.py
GenZ/LLM_inference/utils.py
GenZ/Models/__init__.py
GenZ/Models/attention.py
GenZ/Models/default_models.py
GenZ/Models/embedding.py
GenZ/Models/ffn.py
GenZ/Models/get_language_model.py
GenZ/Models/mamba.py
GenZ/Models/utils.py
GenZ/Models/Model_sets/alibaba.py
GenZ/Models/Model_sets/google.py
GenZ/Models/Model_sets/meta.py
GenZ/Models/Model_sets/microsoft.py
GenZ/Models/Model_sets/misc.py
GenZ/Models/Model_sets/mistral.py
GenZ/Models/Model_sets/nvidia.py
GenZ/utils/plot_rooflines.py
GenZ_paper_charts/v1/variables.py
GenZ_paper_charts/v2/variables.py
Systems/system_configs.py
genz_llm.egg-info/PKG-INFO
genz_llm.egg-info/SOURCES.txt
genz_llm.egg-info/dependency_links.txt
genz_llm.egg-info/requires.txt
genz_llm.egg-info/top_level.txt
tests/test_decode_time.py
tests/test_divide_npu_counts.py
tests/test_model_df_analysis.py
tests/test_model_generations.py
tests/test_prefill_time.py