JAX swMATH ID: 32996 Software Authors: James Bradbury, Roy Frostig, Peter Hawkins, Matthew James Johnson, Chris Leary, Dougal Maclaurin, Skye Wanderman-Milne Description: JAX: composable transformations of Python+NumPy programs. Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more. Homepage: https://jax.readthedocs.io/en/latest/ Source Code: https://github.com/google/jax Dependencies: Python Related Software: TensorFlow; PyTorch; NumPy; Python; GitHub; Adam; ImageNet; DiffSharp; SciPy; Autograd; Theano; PyMC; Matplotlib; GPyTorch; neural-tangents; Chebfun; DeepONet; Julia; CuPy; Scikit Cited in: 25 Publications all top 5 Cited by 96 Authors 3 Perdikaris, Paris G. 2 Schoenholz, Samuel S. 2 Wang, Hanwen 2 Wang, Sifan 1 Abate, Alessandro 1 Adams, Nikolaus A. 1 Akyildiz, Ömer Deniz 1 Anderson, Lara B. 1 Bahri, Yasaman 1 Balog, Matej 1 Bao, Junwei Lucas 1 Barekatain, Mohammadamin 1 Bezgin, Deniz A. 1 Biedenkapp, André 1 Bobev, Nokolai 1 Boureima, I. 1 Bui-Thanh, Tan 1 Calandra, Roberto 1 Charlier, Benjamin 1 Chen, Yuming 1 Coecke, Bob 1 Collin, François-David 1 Cubuk, Ekin D. 1 de Felice, Giovanni 1 Duffin, Connor 1 Durif, Ghislain 1 Eimer, Theresa 1 Fan, Yuwei 1 Faust, Aleksandra 1 Fawzi, Alhussein 1 Feydy, Jean 1 Fischbacher, Thomas 1 Fokina, Daria 1 Francois, Marianne M. 1 Gautason, Fridrik Freyr 1 Gerdes, Mathis 1 Giacobbe, Mirco 1 Giesen, Joachim 1 Girolami, Mark A. 1 Glaunès, Joan Alexis 1 Gray, James A. 1 Gyrya, Vitaliy T. 1 Hassabis, Demis 1 Huang, Aja 1 Hubert, Thomas 1 Hutter, Frank 1 Iliev, Oleg P. 1 Johnston, Hunter 1 Kissas, Georgios 1 Kohli, Pushmeet 1 Krippendorf, Sven 1 Kurien, Susan 1 Laue, Sören 1 Lee, Jaehoon 1 Li, David S. 1 Li, Zhihan 1 Lindauer, Marius 1 Miao, Yingjie 1 Mitterreiter, Matthias 1 Mortari, Daniele 1 Murphy, Kevin P. 1 Novak, Roman 1 Oseledets, Ivan Valer’evich 1 Parker-Holder, Jack 1 Pennington, Jeffrey S. 1 Pilch, Krzysztof 1 Raghuram, Nikhil 1 Rajan, Raghu 1 Romera-Paredes, Bernardino 1 Roy, Diptarko 1 Ruehle, Fabian 1 Ruiz, Francisco J. R. 1 Sabanis, Sotirios 1 Sacks, Michael S. 1 Saenz, Juan A. 1 Sahli Costabal, Francisco 1 Sanz-Alonso, Daniel 1 Schmidt, Steffen J. 1 Schrittwieser, Julian 1 Silver, David M. 1 Sohl-Dickstein, Jascha 1 Song, Xingyou 1 Srivastava, Ankit 1 Sukumar, Natarajan 1 Świrszcz, Grzegorz M. 1 Tac, Vahidullah 1 Teng, Chong 1 Tepole, Adrian B. 1 Toumi, Alexis 1 Tristan, Jean-Baptiste 1 Willett, Rebecca M. 1 Xiao, Lechao 1 Yang, Yibo 1 Ying, Lexing 1 Zhang, Baohe 1 Zhang, Wenbo all top 5 Cited in 15 Serials 5 Computer Methods in Applied Mechanics and Engineering 2 Journal of Computational Physics 2 Journal of High Energy Physics 2 Journal of Statistical Mechanics: Theory and Experiment 1 Journal of Computational and Applied Mathematics 1 Journal of Scientific Computing 1 The Journal of Artificial Intelligence Research (JAIR) 1 Journal of Mathematical Chemistry 1 Fundamenta Informaticae 1 Journal of Machine Learning Research (JMLR) 1 Multiscale Modeling & Simulation 1 Nature, London 1 SIAM/ASA Journal on Uncertainty Quantification 1 Adaptive Computation and Machine Learning 1 SIAM Journal on Mathematics of Data Science all top 5 Cited in 16 Fields 14 Computer science (68-XX) 9 Numerical analysis (65-XX) 4 Partial differential equations (35-XX) 4 Statistics (62-XX) 3 Quantum theory (81-XX) 3 Biology and other natural sciences (92-XX) 2 Probability theory and stochastic processes (60-XX) 2 Mechanics of deformable solids (74-XX) 2 Fluid mechanics (76-XX) 2 Statistical mechanics, structure of matter (82-XX) 2 Relativity and gravitational theory (83-XX) 1 Linear and multilinear algebra; matrix theory (15-XX) 1 Category theory; homological algebra (18-XX) 1 Several complex variables and analytic spaces (32-XX) 1 Harmonic analysis on Euclidean spaces (42-XX) 1 Geophysics (86-XX) Citations by Year