main
alpa
Repository badge with BMinus rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
B
benchmark\megatron\benchmark_mlp_one_case.py
10 1 18
B
tests\pipeline_parallel\test_cross_mesh_resharding.py
36 5 15
B
benchmark\megatron\benchmark_transformer_layer_one_case.py
7 1 21
B
alpa\mesh_profiling.py
151 4 6
B
build_jaxlib\release\generate_pypi_index.py
20 1 13
B
tests\pipeline_parallel\test_stage_construction_util.py
20 5 25
C
tests\shard_parallel\test_moe.py
22 3 37
C
docs\conf.py
6 2 33
C
alpa\util.py
307 17 5
C
examples\llm_serving\scripts\step_2_consolidate_992_shards_to_singleton.py
93 1 36
C
playground\auto_sharding_solver\solver.py
79 1 34
C
playground\auto_sharding_solver\test_solver_attention.py
21 1 37
C
playground\auto_sharding_solver\test_solver_mlp.py
32 1 34
C
benchmark\cupy\profile_communication.py
35 5 31
C
playground\jax_basic\test_matmul_pmap.py
14 1 28
C
alpa\device_mesh.py
537 23 4
C
benchmark\deepspeed\pretrain_gpt2.py
10 1 30
C
playground\pipeline\test_acc_grad.py
49 1 35
C
playground\pipeline\test_compile_and_profile.py
6 1 31
C
playground\xla_builder\test_xla_builder.py
24 1 31