×

JAX

swMATH ID: 32996
Software Authors: Bradbury, James; Frostig, Roy; Hawkins, Peter; Johnson, Matthew James; Leary, Chris; Maclaurin, Dougal; Wanderman-Milne, Skye
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: GitHub; PyTorch; Adam; TensorFlow; Python; NumPy; SciPy; DeepONet; DiffSharp; torchdiffeq; DeepXDE; Matplotlib; DGM; Theano; Autograd; PINNsNTK; hp-VPINNs; JAXopt; FEniCS; ImageNet
Cited in: 91 Documents
all top 5

Cited by 279 Authors

8 Perdikaris, Paris G.
5 Wang, Sifan
3 Adams, Nikolaus A.
3 Bui-Thanh, Tan
3 Karniadakis, George Em
3 Liu, Wing Kam
3 Park, Chanwook
3 Pauwels, Edouard Jean Robert
3 Wagner, Gregory J.
2 Bezgin, Deniz A.
2 Bolte, Jérôme
2 Chen, Yuming
2 Dalton, David R.
2 Gao, Hao
2 Guo, Jiachen
2 Howard, Amanda A.
2 Husmeier, Dirk
2 Lisser, Abdel
2 Lu, Ye
2 Mojumder, Satyajit
2 Nguyen, Hai V.
2 Okuno, Takayuki
2 Oseledets, Ivan Valer’evich
2 Pardo, David
2 Perego, Mauro
2 Saha, Sourav
2 Sanz-Alonso, Daniel
2 Schoenholz, Samuel S.
2 Stinis, Panos
2 Takeda, Akiko
2 Wang, Hanwen
2 Willett, Rebecca M.
2 Wu, Dawen
2 Xue, Tianju
1 Abate, Alessandro
1 Abbott, Trevor
1 Abdussalam, Shehu S.
1 Adami, Stefan
1 Akyildiz, Ömer Deniz
1 Amos, Brandon D.
1 Anderson, Lara B.
1 Apley, Daniel W.
1 Arahata, Shun
1 Asama, Hajime
1 Bahri, Yasaman
1 Balog, Matej
1 Bao, Junwei Lucas
1 Barbastathis, George
1 Barekatain, Mohammadamin
1 Bartman, Piotr
1 Bartosz, Krzysztof
1 Besard, Tim
1 Bhouri, Mohamed Aziz
1 Biedenkapp, André
1 Bobev, Nokolai
1 Bouchard-Côté, Alexandre
1 Boureima, Ismael
1 Braatz, Richard D.
1 Brandstetter, Johannes
1 Brefeld, Ulf
1 Bressan, Stéphane
1 Briol, François-Xavier
1 Buhendwa, Aaron B.
1 Calandra, Roberto
1 Califano, Federico
1 Cao, Jian
1 Carlini, Nicholas
1 Casagrande, Heitor P.
1 Chen, Yifan
1 Churavy, Valentin
1 Coecke, Bob
1 Constantinescu, Emil M.
1 Corander, Jukka
1 Crinò, Chiara
1 Cubuk, Ekin D.
1 da Silva Torres, Ricardo
1 de Felice, Giovanni
1 Duffin, Connor
1 East, Sebastian
1 Edelman, Alan Stuart
1 Eimer, Theresa
1 Erbin, Harold
1 Exl, Lukas
1 Fadel, Samuel G.
1 Fan, Yuwei
1 Faust, Aleksandra
1 Fawzi, Alhussein
1 Fedkiw, Ronald P.
1 Fekete, Imre
1 Fırat, Atakan Hilmi
1 Fischbacher, Thomas
1 Fokina, Daria
1 Francois, Marianne M.
1 Galletti, Gianluca
1 Gan, Zhengtao
1 Gangloff, Hugo
1 Gautason, Fridrik Freyr
1 Geist, A. René
1 Gelman, Andrew
1 Gerdes, Mathis
...and 179 more Authors
all top 5

Cited in 42 Serials

14 Computer Methods in Applied Mechanics and Engineering
13 Journal of Computational Physics
6 Computer Physics Communications
4 Computational Mechanics
4 Journal of High Energy Physics
3 SIAM Journal on Optimization
2 Computers & Mathematics with Applications
2 Computational Optimization and Applications
2 Foundations of Computational Mathematics
2 SIAM Journal on Applied Dynamical Systems
2 Journal of Statistical Mechanics: Theory and Experiment
2 Electronic Journal of Statistics
1 Computers and Fluids
1 Inverse Problems
1 Physica A
1 Information Sciences
1 Journal of Computational and Applied Mathematics
1 Mathematics and Computers in Simulation
1 Results in Mathematics
1 Statistical Science
1 Journal of Scientific Computing
1 Computational Statistics and Data Analysis
1 SIAM Journal on Scientific Computing
1 The Journal of Artificial Intelligence Research (JAIR)
1 Journal of Mathematical Chemistry
1 European Journal of Control
1 International Journal of Computational Fluid Dynamics
1 Chaos
1 European Journal of Mechanics. A. Solids
1 Fundamenta Informaticae
1 Nonlinear Analysis. Real World Applications
1 Journal of Machine Learning Research (JMLR)
1 Multiscale Modeling & Simulation
1 GAMM-Mitteilungen
1 Nature, London
1 AStA. Advances in Statistical Analysis
1 The Annals of Applied Statistics
1 Foundations and Trends in Machine Learning
1 SIAM/ASA Journal on Uncertainty Quantification
1 AMM. Applied Mathematics and Mechanics. (English Edition)
1 Adaptive Computation and Machine Learning
1 SIAM Journal on Mathematics of Data Science

Citations by Year