I Started Learning JAX as a Senior ML Engineer - Here's My First Impression.

SERIES: Learning RL and JAX in Public - from zero to DeepMind ! I have been a machine learning engineer for six years. PyTorch is basically muscle memory at this point. So when I decided to seriously ...