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