JAX: Autograd and XLA https://github.com/google/jax Python, Machine learning, Artificial neural network https://youtube.com/watch?v=WdTeDXsOSj4 Articles PyTorch is dead. Long live JAX.