0.2.1: Increment version number for JAX-CFD

I would like to issue a new release to fix issues like
https://github.com/google/jax-cfd/issues/226

PiperOrigin-RevId: 632484011