README.md
setup.py
jax_cfd/__init__.py
jax_cfd.egg-info/PKG-INFO
jax_cfd.egg-info/SOURCES.txt
jax_cfd.egg-info/dependency_links.txt
jax_cfd.egg-info/requires.txt
jax_cfd.egg-info/top_level.txt
jax_cfd/base/__init__.py
jax_cfd/base/advection.py
jax_cfd/base/advection_test.py
jax_cfd/base/array_utils.py
jax_cfd/base/array_utils_test.py
jax_cfd/base/diffusion.py
jax_cfd/base/diffusion_test.py
jax_cfd/base/equations.py
jax_cfd/base/equations_test.py
jax_cfd/base/fast_diagonalization.py
jax_cfd/base/fast_diagonalization_test.py
jax_cfd/base/fft.py
jax_cfd/base/finite_differences.py
jax_cfd/base/finite_differences_test.py
jax_cfd/base/forcings.py
jax_cfd/base/forcings_test.py
jax_cfd/base/funcutils.py
jax_cfd/base/funcutils_test.py
jax_cfd/base/grids.py
jax_cfd/base/grids_test.py
jax_cfd/base/initial_conditions.py
jax_cfd/base/initial_conditions_test.py
jax_cfd/base/interpolation.py
jax_cfd/base/interpolation_test.py
jax_cfd/base/pressure.py
jax_cfd/base/pressure_test.py
jax_cfd/base/resize.py
jax_cfd/base/resize_test.py
jax_cfd/base/spectral.py
jax_cfd/base/subgrid_models.py
jax_cfd/base/subgrid_models_test.py
jax_cfd/base/test_util.py
jax_cfd/base/validation_problems.py
jax_cfd/base/validation_test.py
jax_cfd/data/__init__.py
jax_cfd/data/evaluation.py
jax_cfd/data/visualization.py
jax_cfd/data/visualization_test.py
jax_cfd/data/xarray_utils.py
jax_cfd/data/xarray_utils_test.py
jax_cfd/ml/__init__.py
jax_cfd/ml/advections.py
jax_cfd/ml/decoders.py
jax_cfd/ml/diffusions.py
jax_cfd/ml/encoders.py
jax_cfd/ml/equations.py
jax_cfd/ml/equations_test.py
jax_cfd/ml/forcings.py
jax_cfd/ml/interpolations.py
jax_cfd/ml/layers.py
jax_cfd/ml/layers_test.py
jax_cfd/ml/layers_util.py
jax_cfd/ml/layers_util_test.py
jax_cfd/ml/model_builder.py
jax_cfd/ml/model_utils.py
jax_cfd/ml/networks.py
jax_cfd/ml/nonlinearities.py
jax_cfd/ml/optimizer_modules.py
jax_cfd/ml/physics_specifications.py
jax_cfd/ml/pressures.py
jax_cfd/ml/tiling.py
jax_cfd/ml/time_integrators.py
jax_cfd/ml/towers.py
jax_cfd/ml/towers_test.py
jax_cfd/ml/viscosities.py