main
alpa
Repository badge with BMinus rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
C
playground\auto_sharding_solver\test_solver_attention.py
21 1 37
C
playground\auto_sharding_solver\test_solver_mlp.py
32 1 34
A
alpa\pipeline_parallel\resharding_tensor.py
43 1 -
A
benchmark\alpa\suite_wresnet.py
6 1 1
B
playground\jax_basic\test_device_put.py
28 1 23
A
playground\jax_basic\test_flop_count.py
1 1 4
A
alpa\collective\collective_group\gloo_collective_group.py
72 1 1
B
benchmark\alpa\util.py
23 1 15
A
playground\jax_basic\test_jit.py
3 1 2
C
playground\jax_basic\test_matmul_pmap.py
14 1 28
A
playground\jax_basic\test_memory_allocator.py
2 1 4
A
playground\jax_basic\test_mixed_precision.py
3 1 3
A
benchmark\cupy\profile_matmul.py
11 1 3
D
playground\jax_basic\test_pjit.py
30 1 47
A
playground\jax_basic\test_pmap.py
3 1 3
A
benchmark\deepspeed\benchmark_gpt2.py
9 1 7
A
playground\jax_basic\test_scan.py
3 1 7
A
playground\jax_basic\test_sharding_spec.py
20 1 8
B
benchmark\deepspeed\benchmark_moe.py
11 1 22
A
playground\jax_basic\test_tuple_args.py
2 1 2