| Pytorch: An imperative style, high-performance deep learning library A Paszke, S Gross, F Massa, A Lerer, J Bradbury, G Chanan, T Killeen, ... Advances in neural information processing systems 32, 2019 | 65444 | 2019 |
| {TensorFlow}: a system for {Large-Scale} machine learning M Abadi, P Barham, J Chen, Z Chen, A Davis, J Dean, M Devin, ... 12th USENIX symposium on operating systems design and implementation (OSDI�…, 2016 | 42810* | 2016 |
| TensorFlow: Large-scale machine learning on heterogeneous systems M Abadi, A Agarwal, P Barham, E Brevdo, Z Chen, C Citro, GS Corrado, ... | 22132 | 2015 |
| Advances in neural information processing systems A Paszke, S Gross, F Massa, A Lerer, J Bradbury, G Chanan, T Killeen, ... Curran Associates, Inc 32, 8024-8035, 2019 | 2763 | 2019 |
| Tensorflow: Large-scale machine learning on heterogeneous distributed systems. arXiv 2016 M Abadi, A Agarwal, P Barham, E Brevdo, Z Chen, C Citro, GS Corrado, ... arXiv preprint arXiv:1603.04467 10, 2019 | 784 | 2019 |
| Pytorch: An imperative style, high-performance deep learning library B Steiner, Z DeVito, S Chintala, S Gross, A Paske, F Massa, A Lerer, ... | 735 | 2019 |
| Device placement optimization with reinforcement learning A Mirhoseini, H Pham, QV Le, B Steiner, R Larsen, Y Zhou, N Kumar, ... International conference on machine learning, 2430-2439, 2017 | 598 | 2017 |
| TensorFlow: large-scale machine learning on heterogeneous distributed systems (2016) M Abadi, A Agarwal, P Barham, E Brevdo, Z Chen, C Citro, GS Corrado, ... arXiv preprint arXiv:1603.04467 52, 2015 | 587 | 2015 |
| Pytorch: An imperative style, high-performance deep learning library. arXiv 2019 A Paszke, S Gross, F Massa, A Lerer, J Bradbury, G Chanan, T Killeen, ... arXiv preprint arXiv:1912.01703 10, 1912 | 410 | 1912 |
| TensorFlow: Large-scale machine learning on heterogeneous systems (2015), software available from tensorflow. org M Abadi, A Agarwal, P Barham, E Brevdo, Z Chen, C Citro, GS Corrado, ... | 370 | 2019 |
| Learning to optimize halide with tree search and random programs A Adams, K Ma, L Anderson, R Baghdadi, TM Li, M Gharbi, B Steiner, ... ACM Transactions on Graphics (TOG) 38 (4), 1-12, 2019 | 335 | 2019 |
| 12th USENIX symposium on operating systems design and implementation (OSDI 16) M Abadi, P Barham, J Chen, Z Chen, A Davis, J Dean, M Devin, ... Savannah, GA, 265-283, 2016 | 313 | 2016 |
| Measuring faithfulness in chain-of-thought reasoning T Lanham, A Chen, A Radhakrishnan, B Steiner, C Denison, ... arXiv preprint arXiv:2307.13702, 2023 | 257 | 2023 |
| Studying large language model generalization with influence functions R Grosse, J Bae, C Anil, N Elhage, A Tamkin, A Tajdini, B Steiner, D Li, ... arXiv preprint arXiv:2308.03296, 2023 | 251 | 2023 |
| Large-scale machine learning on heterogeneous distributed systems M Abadi, AABP TensorFlow Proceedings of the 12th USENIX symposium on operating systems design and�…, 2016 | 245 | 2016 |
| A hierarchical model for device placement A Mirhoseini, A Goldie, H Pham, B Steiner, QV Le, J Dean International conference on learning representations, 2018 | 220 | 2018 |
| TensorFlow: Large-scale machine learning on heterogeneous systems. arXiv 2015 M Abadi, A Agarwal, P Barham, E Brevdo, Z Chen, C Citro, GS Corrado, ... arXiv preprint arXiv:1603.04467, 2016 | 180 | 2016 |
| PyTorch: An Imperative Style, High-Performance Deep Learning Library, Dec A Paszke, S Gross, F Massa, A Lerer, J Bradbury, G Chanan, T Killeen, ... arXiv preprint arXiv:1912.01703, 2019 | 144 | 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 | 132 | 2022 |
| TensorFlow M Abadi, A Agarwal, P Barham, E Brevdo, Z Chen, C Citro, GS Corrado, ... Large-scale machine learning on heterogeneous systems 11, 2015 | 111 | 2015 |
