code-first-ml

Code First ML

These books provide a code-first introduction to ML.

Code First ML Book 1

Code First ML Book 2

Main Toolset

  1. JAX/PyTorch
  2. TFP for distributions
  3. BlackJax for MCMC
  4. Optax and Jaxopt for optimization
  5. Flax for neural networks