google/jax-cfd issues and pull requests

#239 - Bugfix: vorticity_space_forcing was applying jnp.fft.rfft2 to GridVariable when array was expected.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#238 - Re-enable async dispatch on JAX CPU in equations_test.py

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#237 - Avoid complex->real casts via jax.numpy.astype

Pull Request - State: closed - Opened by copybara-service[bot] 2 months ago

#236 - Avoid complex->real casts via jax.numpy.astype

Pull Request - State: open - Opened by copybara-service[bot] 2 months ago

#235 - [numpy] Fix users of NumPy APIs that are removed in NumPy 2.0.

Pull Request - State: closed - Opened by copybara-service[bot] 2 months ago

#234 - Pressure Variable

Issue - State: open - Opened by chenwydj 3 months ago

#233 - Examples of using kolmogorov_forcing in semi_implicit_navier_stokes

Issue - State: open - Opened by chenwydj 3 months ago

#232 - This statement 'self_advection.advection_module = @modular_self_advection' does not work, as well as the result is inconsistent with the paper

Issue - State: open - Opened by yixin-cfd 4 months ago - 2 comments

#231 - Why is there no training code? How about training other examples? Like square cavity flow?

Issue - State: open - Opened by yixin-cfd 4 months ago

#230 - Fixed Taylor Green decay coefficient and added an example

Pull Request - State: open - Opened by scaomath 5 months ago - 2 comments

#229 - Replace deprecated `jax.tree_` functions with `jax.tree.`

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#228 - Increment version number for JAX-CFD

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#227 - Increase atol in SemiImplicitNavierStokesTest.test_divergence_and_momentum_semi_implicit_sinusoidal_velocity_base to reduce flakes

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#226 - AttributeError: module 'jax.numpy' has no attribute 'DeviceArray'

Issue - State: closed - Opened by FMGS666 6 months ago - 3 comments

#225 - Disable async dispatch on JAX CPU for a jax_cfd test, i.e., opt out of the change introduced in https://github.com/google/jax/pull/20470, as it is causing some real failures under tsan.

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago

#224 - Add extra field variables as output for the step function

Issue - State: closed - Opened by scaomath 6 months ago - 1 comment

#223 - Allowing for 2/3 rule smoothing for non square grids

Pull Request - State: closed - Opened by clearya1 6 months ago
Labels: pull ready

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

Pull Request - State: open - Opened by Edward-Sun 7 months ago - 1 comment

#221 - ci: Use GITHUB_OUTPUT envvar instead of set-output command

Pull Request - State: open - Opened by arunsathiya 9 months ago - 1 comment

#220 - skip seed-sensitive random test

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#219 - relax test tolerances that failed with new random values

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#218 - Jax.devicearray is no longer available in recent Jax versions

Issue - State: closed - Opened by hashimmg 11 months ago

#217 - Updated jax.config import

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago

#216 - Updated jax.config import

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago

#215 - [LSC] Ignore incorrect type annotations related to jax.numpy APIs

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#214 - Tests fail upon fresh install.

Issue - State: open - Opened by danielkelshaw about 1 year ago - 1 comment

#213 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#212 - Exception has occurred: KeyError 'experimental_xmap_spmd_lowering'

Issue - State: open - Opened by MCSchulz-git over 1 year ago

#211 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#210 - Remove hk.experimental.named_call.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#209 - Colab

Issue - State: open - Opened by gwonchan over 1 year ago

#207 - AttributeError: AxesImage.set() got an unexpected keyword argument 'msg'

Issue - State: closed - Opened by JFGoes over 1 year ago

#206 - Remove references to deprecated jax.ShapedArray

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#205 - Fix pytype failures related to teaching pytype about NumPy scalar types.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#204 - Suppress some pytype errors related to improved JAX types under pytype.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#203 - Replaces references to jax.numpy.DeviceArray with jax.Array.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#202 - Suppress some pytype errors related to jnp.DeviceArray == jax.Array.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#201 - trajectory function use target_trajectory as input in loss_and_gradient

Issue - State: open - Opened by gwen-git over 1 year ago

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

Issue - State: open - Opened by wagenaartje over 1 year ago - 3 comments

#199 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#198 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago - 1 comment

#197 - Modelling turbulent channel flow

Issue - State: closed - Opened by diegodelc over 1 year ago

#196 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#195 - Allows for nonperiodic subgrid models.

Pull Request - State: open - Opened by copybara-service[bot] over 1 year ago - 1 comment

#194 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#193 - Satisfy stricter input requirement for `jnp.linalg.norm`

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#192 - Satisfy stricter input requirement for `jnp.linalg.norm`

Pull Request - State: open - Opened by copybara-service[bot] almost 2 years ago

#191 - Add `jax.Array` public type which replaces `jnp.DeviceArray` and `jax.core.Tracer`.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago