DeepMind releases JAX neural networks libraries · HackerTrans