pytorch/benchmarks/nested
..
nested_bmm_bench.py