ColossalAI/colossalai/cli/benchmark
..
__init__.py
benchmark.py
models.py
utils.py