Joe Schoonover

Hey @BigShuiTai - We're currently looking into this as well . At the moment, we're starting here with [Jax on ROCm](https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/jax-install.html) . I'll keep you posted on what we find...

@BigShuiTai - this is awesome. Curious why you're setting `PYTORCH_ROCM_ARCH`. My understanding is that pytorch is not a dependency here. @garrettbyrd check this out

> It would make sense to add some simple tests for each of the new BCs in 2D and 3D to satisfy codecov. > > This would likely be under...

Note for this issue and for the release notes --- We have not explicitly tested meshes with flip=1 in 2-D and flip>=1 in 3-D, though these bits are coded up...

Hey @janhclem ! Thanks for sharing notes and data. I'll work on making sure we can ingest this data first with the diagnostics values. Once we can get something working...