jax-cfd icon indicating copy to clipboard operation
jax-cfd copied to clipboard

Computational Fluid Dynamics in JAX

Results 52 jax-cfd issues
Sort by recently updated
recently updated
newest added

Any plan to move to colab?

Hi, I'm trying to understand the logic behind the various functions in ml.train_utils. in the definition of loss_and_gradient the returned _loss function uses the target trajectory as part of the...

Satisfy stricter input requirement for `jnp.linalg.norm` The previous version of the code caused an error under the new more restrictive API of `jnp.linalg.norm` (see https://github.com/google/jax/pull/12670).

How to get the simulated pressure variable along the trajectory? For example in this notebook: https://github.com/google/jax-cfd/blob/main/notebooks/demo.ipynb Thank you very much!

Dear researchers: Do you happen to have any example of using [kolmogorov_forcing](https://github.com/google/jax-cfd/blob/main/jax_cfd/base/forcings.py#L63) in [semi_implicit_navier_stokes](https://github.com/google/jax-cfd/blob/main/jax_cfd/base/equations.py#L128) ? Naively passing `kolmogorov_forcing` into `semi_implicit_navier_stokes` will not work. Mimic the usage in [spectral](https://github.com/google/jax-cfd/blob/main/jax_cfd/spectral/equations.py#L213) will give...

![image](https://github.com/google/jax-cfd/assets/162121225/52c20a8c-aeba-4cef-b5e9-9e39bf28b12b) ![image](https://github.com/google/jax-cfd/assets/162121225/78339f52-6df2-4f1e-bc1e-df6ee0e05fad) ![image](https://github.com/google/jax-cfd/assets/162121225/e701f743-2d28-4b4a-9ca4-d6b82adba1f3) ![Figure_2](https://github.com/google/jax-cfd/assets/162121225/ac5d53c4-2bb8-461c-9507-9ba4c8d83325) ![Figure_3](https://github.com/google/jax-cfd/assets/162121225/a20cbcd6-6950-4b94-9216-40f59b206df0)

The scale for the Taylor Green example should be $\exp(-(k_x^2 +k_y^2) \nu t )$, otherwise the factor coming out from $\partial_t \mathbf{u}$ cannot be canceled with $\nu\Delta \mathbf{u}$. An example...