Pinned Loading

1. SamSz04.github.io SamSz04.github.io Public

   JavaScript

2. tpu_module_test tpu_module_test Public

   JAX-based. Test and evaluate op's results on different hardware including TPUs and GPUs

   Python

3. Models4Fusion Models4Fusion Public

   Model zoo for generating Hlo dataset and further experiment on RL fusion.

   Python 5

4. xla_fusion xla_fusion Public

   Support beam search in the XLA framework, and we can compare the performance of beam search and XLA's priority fusion.

   C++ 1

5. autocomp-tpu-guide autocomp-tpu-guide Public

6. go_fusion go_fusion Public

   Python