view article Article On Learning JAX – A Framework for High Performance Machine Learning Dec 3, 2023 • 3