Code First ML These books provide a code-first introduction to ML. Code First ML Book 1 Code First ML Book 2 Main Toolset JAX/PyTorch TFP for distributions BlackJax for MCMC Optax and Jaxopt for optimization Flax for neural networks