Numpy on the GPU? Speeding up Simple Machine Learning Algorithms with JAX
Demo: JAX, Flax and Gemma
JAX: Accelerated Machine Learning Research via Composable Function Transformations in Python
JAX: accelerated machine learning research via composable function transformations in Python
*Fast* Python and more - Functional languages for machine learning
Physics-Informed Neural Networks in JAX (with Equinox & Optax)
JAX: Accelerated Machine Learning Research | SciPy 2020 | VanderPlas
PyTorch vs TensorFlow | Ishan Misra and Lex Fridman
