Closed mhechthz closed 12 months ago
Hi, I think it should be relatively straightforward to translate it since JAX essentially implements the numpy API. The three points that need changes are:
jax.numpy as jnp
to numpy as np
and subsequently change all jnp
to np
Would be a nice thing if you could give it a try and open a PR 😊
Could you add the lattice boltzmann simulation without the use of jax?