JAX is a machine learning framework by Google.
The core of JAX handles things like calculating gradients or vectorizing operations. But there are libraries called Flax and Flax-Linen that implement neural networks on top of JAX.
JAX
Reading time: less than 1 minute
The following pages link here
Citation
If you find this work useful, please cite it as:@article{yaltirakli, title = "JAX", author = "Yaltirakli, Gokberk", journal = "gkbrk.com", year = "2025", url = "https://www.gkbrk.com/jax" }
Not using BibTeX? Click here for more citation styles.
IEEE Citation
Gokberk Yaltirakli, "JAX", February, 2025. [Online]. Available: https://www.gkbrk.com/jax. [Accessed Feb. 04, 2025].
APA Style
Yaltirakli, G. (2025, February 04). JAX. https://www.gkbrk.com/jax
Bluebook Style
Gokberk Yaltirakli, JAX, GKBRK.COM (Feb. 04, 2025), https://www.gkbrk.com/jax
