v0.6.2

New features:

• nt.stax.repeat layer allowing fast compilation of very deep networks (see #168 and thanks @jglaser!)
• Add a Colab notebook accompanying Precise Learning Curves and Higher-Order Scaling Limits for Dot Product Kernel Regression

Improvements:

• Support M1 macs (see #171)
• Internal changes to support newer versions of JAX

Breaking changes:

• Drop support for Python 3.7 following JAX and tf2jax version policy
• On GPU, require CuDNN>=8.2 to support JAX 0.4.3