Grokking Loss Curves
Initializing jax-js...