Put on your metaphorical safety goggles and start building something weird with JAX.| machine learning musings
