README.md
pyproject.toml
tests/test_chorin_algebraic_aux_jax.py
tests/test_dt_max_wave_free_jax_req190.py
tests/test_elliptic_bc_jax_req174.py
tests/test_stage_list_jax_req173.py
tests/test_swe_2d_jax.py
tests/test_swe_dambreak_jax.py
tests/test_time_position_operators_jax_req185.py
tests/test_uniform_rank_malpasset_req84.py
tests/test_userfunctions_jax.py
zoomy_jax/LICENSE
zoomy_jax/README.md
zoomy_jax/__init__.py
zoomy_jax/gpurank.py
zoomy_jax.egg-info/PKG-INFO
zoomy_jax.egg-info/SOURCES.txt
zoomy_jax.egg-info/dependency_links.txt
zoomy_jax.egg-info/requires.txt
zoomy_jax.egg-info/top_level.txt
zoomy_jax/fvm/__init__.py
zoomy_jax/fvm/eigvals.py
zoomy_jax/fvm/flux.py
zoomy_jax/fvm/halo_exchange_jax.py
zoomy_jax/fvm/implicit_ad.py
zoomy_jax/fvm/jvp_jax.py
zoomy_jax/fvm/march_emitted.py
zoomy_jax/fvm/nonconservative_flux.py
zoomy_jax/fvm/observers.py
zoomy_jax/fvm/ode.py
zoomy_jax/fvm/precice_solver.py
zoomy_jax/fvm/reconstruction_jax.py
zoomy_jax/fvm/solver_chorin_vam_jax.py
zoomy_jax/fvm/solver_imex_jax.py
zoomy_jax/fvm/solver_jax.py
zoomy_jax/fvm/spmd_jax.py
zoomy_jax/fvm/userfunctions.py
zoomy_jax/fvm/solver2/__init__.py
zoomy_jax/fvm/solver2/blocks.py
zoomy_jax/fvm/solver2/context.py
zoomy_jax/fvm/solver2/march.py
zoomy_jax/fvm/solver2/state.py
zoomy_jax/gnn_blueprint/README.md
zoomy_jax/gnn_blueprint/__init__.py
zoomy_jax/gnn_blueprint/benchmark_gmres_guess.py
zoomy_jax/gnn_blueprint/benchmark_poisson_iterations.py
zoomy_jax/gnn_blueprint/benchmark_spectral_mesh_suite.py
zoomy_jax/gnn_blueprint/cases_gn_topo.py
zoomy_jax/gnn_blueprint/common.py
zoomy_jax/gnn_blueprint/customize_from_collect.py
zoomy_jax/gnn_blueprint/dataset_2d_mg_multibranch.py
zoomy_jax/gnn_blueprint/dataset_deltaq.py
zoomy_jax/gnn_blueprint/demo_fft_gnn_1d.py
zoomy_jax/gnn_blueprint/demo_local_experts_transient.py
zoomy_jax/gnn_blueprint/demo_local_gnn.py
zoomy_jax/gnn_blueprint/demo_local_gnn_timeseries.py
zoomy_jax/gnn_blueprint/eval_standalone_predictor.py
zoomy_jax/gnn_blueprint/fourier_1d.py
zoomy_jax/gnn_blueprint/global_coupling.py
zoomy_jax/gnn_blueprint/graph_laplacian_spectral.py
zoomy_jax/gnn_blueprint/graph_mp_multibranch.py
zoomy_jax/gnn_blueprint/mesh_1d_poisson.py
zoomy_jax/gnn_blueprint/mesh_2d_poisson.py
zoomy_jax/gnn_blueprint/mesh_spectral_geometry.py
zoomy_jax/gnn_blueprint/mg_structured_hierarchy.py
zoomy_jax/gnn_blueprint/mg_structured_hierarchy_1d.py
zoomy_jax/gnn_blueprint/models_deltaq.py
zoomy_jax/gnn_blueprint/multigrid_gnn_toy.py
zoomy_jax/gnn_blueprint/nonuniform_spectral.py
zoomy_jax/gnn_blueprint/poisson_1d.py
zoomy_jax/gnn_blueprint/predictor_learned_multilevel.py
zoomy_jax/gnn_blueprint/run_2d_mg_quick_results.py
zoomy_jax/gnn_blueprint/run_full_loop.sh
zoomy_jax/gnn_blueprint/smooth_1d.py
zoomy_jax/gnn_blueprint/train_2d_mg_benchmark.py
zoomy_jax/gnn_blueprint/train_autosearch_deltaq.py
zoomy_jax/gnn_blueprint/train_deltaq.py
zoomy_jax/gnn_blueprint/train_hybrid_pressure_residual.py
zoomy_jax/gnn_blueprint/train_multilevel_fft1d.py
zoomy_jax/gnn_blueprint/train_poisson_arch_benchmark.py
zoomy_jax/gnn_blueprint/train_residual_target_deltaq.py
zoomy_jax/gnn_blueprint/train_spectral_mesh_suite.py
zoomy_jax/gnn_blueprint/train_vcycle_structured_poisson.py
zoomy_jax/gnn_blueprint/train_vcycle_structured_poisson_1d.py
zoomy_jax/gnn_blueprint/vcycle_imex_bridge.py
zoomy_jax/gnn_blueprint/vcycle_structured_gnn.py
zoomy_jax/mesh/__init__.py
zoomy_jax/mesh/mesh.py
zoomy_jax/mesh/partition.py
zoomy_jax/mesh/partition_jax.py
zoomy_jax/misc/__init__.py
zoomy_jax/misc/io.py
zoomy_jax/symsolver/__init__.py
zoomy_jax/symsolver/compile_cache.py
zoomy_jax/symsolver/coupling.py
zoomy_jax/symsolver/hypre.py
zoomy_jax/symsolver/parallel.py
zoomy_jax/symsolver/runtime.py
zoomy_jax/symsolver/solver.py
zoomy_jax/template/__init__.py
zoomy_jax/template/backend.py
zoomy_jax/template/linalg.py
zoomy_jax/transformation/__init__.py
zoomy_jax/transformation/jax_runtime.py
zoomy_jax/transformation/to_jax.py