JAX 3 JAX 101 - Parallel Evaluation in JAX Apr 21, 2024 JAX 101 - Parallel Evaluation in JAX Apr 21, 2024 JAX 101 - Working with Pytrees Apr 21, 2024