jax-cfd
jax-cfd copied to clipboard
Computational Fluid Dynamics in JAX
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...
Allows for nonperiodic subgrid models.
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...
    
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...
Small typo in the docstrings.