jax-cfd

jax-cfd copied to clipboard

Internal change

1

Internal change

copybara-service[bot]

Internal change

1

Internal change

copybara-service[bot]

Internal change

1

Internal change

copybara-service[bot]

Internal change

1

Internal change

copybara-service[bot]

Internal change

copybara-service[bot]

Add TSM-PDE to "Projects using JAX-CFD"

1

Project link: https://github.com/Edward-Sun/TSM-PDE Paper link (ICML): https://arxiv.org/abs/2302.08105

Edward-Sun

ci: Use GITHUB_OUTPUT envvar instead of set-output command

1

`save-state` and `set-output` commands used in GitHub Actions are deprecated and [GitHub recommends using environment files](https://github.blog/changelog/2023-07-24-github-actions-update-on-save-state-and-set-output-commands/). This PR updates the usage of `::set-output` to `"$GITHUB_OUTPUT"` Instructions for envvar usage from...

arunsathiya

Tests fail upon fresh install.

1

Upon following the instructions to setup a development environment, i.e: ```bash git clone https://github.com/google/jax-cfd; cd jax-cfd python -m venv venv source venv/bin/activate pip install --upgrade pip pip install jaxlib pip...

danielkelshaw

Can't run spectral_forced_turbulence due to jnp.linalg.norm error

3

I have copied the code from [spectral_forced_turbulence.ipynb](https://github.com/google/jax-cfd/blob/main/notebooks/spectral_forced_turbulence.ipynb), but it gives the following error: ``` --------------------------------------------------------------------------- UnfilteredStackTrace Traceback (most recent call last) in [/usr/local/lib/python3.8/dist-packages/jax_cfd/base/initial_conditions.py](https://localhost:8080/#) in filtered_velocity_field(rng_key, grid, maximum_velocity, peak_wavenumber, iterations) 110...

wagenaartje

When running layers_util_test.py with jax[CPU] version later than 0.4.1, it throws a key error 'experimental_xmap_spmd_lowering'.

MCSchulz-git