jax-cfd
jax-cfd copied to clipboard
Avoid complex->real casts via jax.numpy.astype
Avoid complex->real casts via jax.numpy.astype
This currently issues a warning about implicitly discarding the imaginary part, and it will issue an error in the future.