jax-cfd

5e31fa1f · mashun1 · 5e31fa1f · 5e31fa1f · 5e31fa1f · 5e31fa1f
Commit 5e31fa1f authored May 23, 2024 by mashun1
20 changed files
--- a/jax_cfd/collocated/diffusion_test.py
+++ b/jax_cfd/collocated/diffusion_test.py
--- a/jax_cfd/collocated/equations.py
+++ b/jax_cfd/collocated/equations.py
--- a/jax_cfd/collocated/equations_test.py
+++ b/jax_cfd/collocated/equations_test.py
--- a/jax_cfd/collocated/initial_conditions.py
+++ b/jax_cfd/collocated/initial_conditions.py
--- a/jax_cfd/collocated/initial_conditions_test.py
+++ b/jax_cfd/collocated/initial_conditions_test.py
--- a/jax_cfd/collocated/pressure.py
+++ b/jax_cfd/collocated/pressure.py
--- a/jax_cfd/collocated/pressure_test.py
+++ b/jax_cfd/collocated/pressure_test.py
--- a/jax_cfd/data/__init__.py
+++ b/jax_cfd/data/__init__.py
--- a/jax_cfd/data/__pycache__/__init__.cpython-310.pyc
+++ b/jax_cfd/data/__pycache__/__init__.cpython-310.pyc
--- a/jax_cfd/data/__pycache__/evaluation.cpython-310.pyc
+++ b/jax_cfd/data/__pycache__/evaluation.cpython-310.pyc
--- a/jax_cfd/data/__pycache__/visualization.cpython-310.pyc
+++ b/jax_cfd/data/__pycache__/visualization.cpython-310.pyc
--- a/jax_cfd/data/__pycache__/visualization_test.cpython-310-pytest-8.2.0.pyc
+++ b/jax_cfd/data/__pycache__/visualization_test.cpython-310-pytest-8.2.0.pyc
--- a/jax_cfd/data/__pycache__/xarray_utils.cpython-310.pyc
+++ b/jax_cfd/data/__pycache__/xarray_utils.cpython-310.pyc
--- a/jax_cfd/data/__pycache__/xarray_utils_test.cpython-310-pytest-8.2.0.pyc
+++ b/jax_cfd/data/__pycache__/xarray_utils_test.cpython-310-pytest-8.2.0.pyc
--- a/jax_cfd/data/evaluation.py
+++ b/jax_cfd/data/evaluation.py
--- a/jax_cfd/data/visualization.py
+++ b/jax_cfd/data/visualization.py
--- a/jax_cfd/data/visualization_test.py
+++ b/jax_cfd/data/visualization_test.py
--- a/jax_cfd/data/xarray_utils.py
+++ b/jax_cfd/data/xarray_utils.py
--- a/jax_cfd/data/xarray_utils_test.py
+++ b/jax_cfd/data/xarray_utils_test.py
--- a/jax_cfd/ml/__init__.py
+++ b/jax_cfd/ml/__init__.py