Fast Autoscheduling for Sparse ML Frameworks

Bobby Yan; Alexander J Root; Trevor Gale; David Broman; Fredrik Kjolstad

doi:10.1109/CGO68049.2026.11394842

CGO · January 2026

Fast Autoscheduling for Sparse ML Frameworks

Bobby Yan Alexander J Root Trevor Gale David Broman Fredrik Kjolstad

IEEE/ACM International Symposium on Code Generation and Optimization

Read PDF DOI

Topics Scheduling & autotuning Sparse tensor algebra

Related software

Scorch: An Optimized Sparse PyTorch A library that adds sparse tensors and JIT-compiled kernels to PyTorch with a compatible API, automating loop ordering, tiling, and format inference for sparse machine-learning workloads.

Cite this paper

BibTeX

@inproceedings{yan2026fast,
  author    = {Yan, Bobby and Root, Alexander J and Gale, Trevor and Broman, David and Kjolstad, Fredrik},
  title     = {Fast Autoscheduling for Sparse {ML} Frameworks},
  booktitle = {2026 IEEE/ACM International Symposium on Code Generation and Optimization (CGO)},
  year      = {2026},
  month     = jan,
  pages     = {28--43},
  publisher = {IEEE},
  doi       = {10.1109/CGO68049.2026.11394842},
  url       = {https://doi.org/10.1109/CGO68049.2026.11394842}
}

Citation

Bobby Yan, Alexander J Root, Trevor Gale, David Broman, and Fredrik Kjolstad. Fast Autoscheduling for Sparse ML Frameworks. CGO, 2026.

All publications