Code First ML

These books provide a code first introduction to ML. These books mirror book 1 and book 2 on Probabilistic ML by Kevin Murphy.

Code First ML Book 1

Code First ML Book 2

Main Toolset

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