Deep Learning with Jax
https://www.manning.com/books/deep-learning-with-jax
#ycombinator #neural_networks #Python #Google #Flax #Haiku #gradient_processing #optimization #numerical #computations #federated_learning #parallelized #distributed #primitives #differentiable