Sparsetir: Composable abstractions for sparse compilation in deep learning Z Ye, R Lai, J Shao, T Chen, L Ceze Proceedings of the 28th ACM International Conference on Architectural …, 2023 | 55 | 2023 |
Tensorir: An abstraction for automatic tensorized program optimization S Feng, B Hou, H Jin, W Lin, J Shao, R Lai, Z Ye, L Zheng, CH Yu, Y Yu, ... Proceedings of the 28th ACM International Conference on Architectural …, 2023 | 42 | 2023 |
Tensor program optimization with probabilistic programs J Shao, X Zhou, S Feng, B Hou, R Lai, H Jin, W Lin, M Masuda, CH Yu, ... Advances in Neural Information Processing Systems 35, 35783-35796, 2022 | 14 | 2022 |
Relax: Composable Abstractions for End-to-End Dynamic Machine Learning R Lai, J Shao, S Feng, SS Lyubomirsky, B Hou, W Lin, Z Ye, H Jin, Y Jin, ... arXiv preprint arXiv:2311.02103, 2023 | 2 | 2023 |
Emerging Platforms Meet Emerging LLMs: A Year-Long Journey of Top-Down Development S Feng, J Liu, R Lai, CF Ruan, Y Yu, L Zhang, T Chen arXiv preprint arXiv:2404.09151, 2024 | | 2024 |