Acme: A research framework for distributed reinforcement learning MW Hoffman, B Shahriari, J Aslanides, G Barth-Maron, N Momchev, ... arXiv preprint arXiv:2006.00979, 2020 | 271 | 2020 |
Haiku: Sonnet for JAX T Hennigan, T Cai, T Norman, I Babuschkin Version 0.0 9, 2020 | 188* | 2020 |
The DeepMind JAX Ecosystem I Babuschkin, K Baumli, A Bell, S Bhupatiraju, J Bruce, P Buchlovsky, ... URL http://github. com/deepmind 24, 25, 2020 | 172* | 2020 |
The DeepMind JAX Ecosystem, 2020 IB DeepMind, K Baumli, A Bell, S Bhupatiraju, J Bruce, P Buchlovsky, ... URL http://github. com/google-deepmind, 0 | 32 | |
The DeepMind JAX Ecosystem IB DeepMind, K Baumli, A Bell, S Bhupatiraju, J Bruce, P Buchlovsky, ... | 22 | 2020 |
Synthesizing optimal parallelism placement and reduction strategies on hierarchical systems for deep learning N Xie, T Norman, D Grewe, D Vytiniotis Proceedings of Machine Learning and Systems 4, 548-566, 2022 | 15 | 2022 |
Automap: Towards ergonomic automated parallelism for ML models M Schaarschmidt, D Grewe, D Vytiniotis, A Paszke, GS Schmid, T Norman, ... arXiv preprint arXiv:2112.02958, 2021 | 12 | 2021 |
Partir: Composing spmd partitioning strategies for machine learning S Alabed, D Belov, B Chrzaszcz, J Franco, D Grewe, D Maclaurin, ... arXiv preprint arXiv:2401.11202, 2024 | 3 | 2024 |
Automatic Discovery of Composite SPMD Partitioning Strategies in PartIR S Alabed, D Grewe, J Franco, B Chrzaszcz, T Natan, T Norman, NA Rink, ... arXiv preprint arXiv:2210.06352, 2022 | | 2022 |