Machine learning at facebook: Understanding inference at the edge CJ Wu, D Brooks, K Chen, D Chen, S Choudhury, M Dukhan, ... 2019 IEEE international symposium on high performance computer architecture …, 2019 | 514 | 2019 |
Compilergym: Robust, performant compiler optimization environments for ai research C Cummins, B Wasti, J Guo, B Cui, J Ansel, S Gomez, S Jain, J Liu, ... 2022 IEEE/ACM International Symposium on Code Generation and Optimization …, 2022 | 56 | 2022 |
Enabling full body AR with Mask R-CNN2Go A Jindal, A Tulloch, B Sharma, B Wasti, F Yang, G Gkioxari, J Kim, ... | 9 | 2018 |
Q-gym: An equality saturation framework for dnn inference exploiting weight repetition C Fu, H Huang, B Wasti, C Cummins, R Baghdadi, K Hazelwood, Y Tian, ... Proceedings of the International Conference on Parallel Architectures and …, 2022 | 4 | 2022 |
LoopStack: a Lightweight Tensor Algebra Compiler Stack B Wasti, JP Cambronero, B Steiner, H Leather, A Zlateski arXiv preprint arXiv:2205.00618, 2022 | 4 | 2022 |
Layer Skip: Enabling Early Exit Inference and Self-Speculative Decoding M Elhoushi, A Shrivastava, D Liskovich, B Hosmer, B Wasti, L Lai, ... arXiv preprint arXiv:2404.16710, 2024 | | 2024 |
LoopTune: Optimizing Tensor Computations with Reinforcement Learning D Grubisic, B Wasti, C Cummins, J Mellor-Crummey, A Zlateski arXiv preprint arXiv:2309.01825, 2023 | | 2023 |
Semisupervised Learning on Heterogeneous Graphs and its Applications to Facebook News Feed C Ju, J Li, B Wasti, S Guo arXiv preprint arXiv:1805.07479, 2018 | | 2018 |
LoopStack: ML-friendly ML Compiler Stack B Wasti, D Grubisic, B Steiner, A Zlateski | | |
Community Infrastructure for Applying Reinforcement Learning to Compiler Optimizations C Cummins, B Wasti, J Guo, B Cui, J Ansel, S Gomez, S Jain, J Liu, ... | | |