LICENSE
MANIFEST.in
README.md
build_config.py
pyproject.toml
setup.py
setup_cuda.py
src/geophyai/__init__.py
src/sweep/_C.py
src/sweep/__init__.py
src/sweep/build.py
src/sweep/cli.py
src/sweep/py.typed
src/sweep/scalars.py
src/sweep/signal.py
src/sweep/backend/__init__.py
src/sweep/backend/c/__init__.py
src/sweep/backend/c/abi.py
src/sweep/backend/c/adapt.py
src/sweep/backend/c/entries.py
src/sweep/backend/c/jit.py
src/sweep/backend/c/loader.py
src/sweep/backend/c/runners.py
src/sweep/backend/jax/__init__.py
src/sweep/backend/jax/cuda.py
src/sweep/backend/torch/__init__.py
src/sweep/backend/torch/binding.py
src/sweep/backend/torch/cuda.py
src/sweep/core/__init__.py
src/sweep/core/arguments.py
src/sweep/core/geometry.py
src/sweep/core/topography.py
src/sweep/core/validation.py
src/sweep/csrc/CMakeLists.txt
src/sweep/csrc/bindings/bindings_utils.h
src/sweep/csrc/bindings/module.cpp
src/sweep/csrc/core/buf.h
src/sweep/csrc/core/buflist.h
src/sweep/csrc/core/capi.h
src/sweep/csrc/core/check.h
src/sweep/csrc/core/device.h
src/sweep/csrc/core/dtype.h
src/sweep/csrc/core/input_core.h
src/sweep/csrc/core/outputs.h
src/sweep/csrc/core/runner.h
src/sweep/csrc/cuda/common/acoustic.h
src/sweep/csrc/cuda/common/acoustic_vrz_fused.cuh
src/sweep/csrc/cuda/common/adapt_inputs.h
src/sweep/csrc/cuda/common/boundary_runtime.cuh
src/sweep/csrc/cuda/common/boundarysaver.cu
src/sweep/csrc/cuda/common/boundarysaver.cuh
src/sweep/csrc/cuda/common/buf_torch.h
src/sweep/csrc/cuda/common/capi.cu
src/sweep/csrc/cuda/common/checkpoint_runtime.cuh
src/sweep/csrc/cuda/common/common.cu
src/sweep/csrc/cuda/common/common.cuh
src/sweep/csrc/cuda/common/context.h
src/sweep/csrc/cuda/common/cudautils.h
src/sweep/csrc/cuda/common/das.h
src/sweep/csrc/cuda/common/das_mu.h
src/sweep/csrc/cuda/common/derived_models.cu
src/sweep/csrc/cuda/common/derived_models.h
src/sweep/csrc/cuda/common/elastic.h
src/sweep/csrc/cuda/common/elastic_free_surface.cuh
src/sweep/csrc/cuda/common/eq_driver.cuh
src/sweep/csrc/cuda/common/layout.cu
src/sweep/csrc/cuda/common/sg_driver.cuh
src/sweep/csrc/cuda/common/boundary/disk_io.cuh
src/sweep/csrc/cuda/common/boundary/kernels.cuh
src/sweep/csrc/cuda/common/boundary/runtime.cuh
src/sweep/csrc/cuda/common/boundary/saver.cuh
src/sweep/csrc/cuda/common/boundary/session.cu
src/sweep/csrc/cuda/common/boundary/session.cuh
src/sweep/csrc/cuda/common/boundary/strip.cuh
src/sweep/csrc/cuda/common/boundary/types.cuh
src/sweep/csrc/cuda/equations/acoustic2d/acoustic2d.h
src/sweep/csrc/cuda/equations/acoustic2d/backward.cu
src/sweep/csrc/cuda/equations/acoustic2d/driver_traits.cuh
src/sweep/csrc/cuda/equations/acoustic2d/forward.cu
src/sweep/csrc/cuda/equations/acoustic2d/kernels.cu
src/sweep/csrc/cuda/equations/acoustic2d/kernels.cuh
src/sweep/csrc/cuda/equations/acoustic3d/acoustic3d.h
src/sweep/csrc/cuda/equations/acoustic3d/backward.cu
src/sweep/csrc/cuda/equations/acoustic3d/driver_traits.cuh
src/sweep/csrc/cuda/equations/acoustic3d/forward.cu
src/sweep/csrc/cuda/equations/acoustic3d/kernels.cu
src/sweep/csrc/cuda/equations/acoustic3d/kernels.cuh
src/sweep/csrc/cuda/equations/acoustic_lsrtm2d/acoustic_lsrtm2d.h
src/sweep/csrc/cuda/equations/acoustic_lsrtm2d/backward.cu
src/sweep/csrc/cuda/equations/acoustic_lsrtm2d/forward.cu
src/sweep/csrc/cuda/equations/acoustic_lsrtm2d/kernels.cu
src/sweep/csrc/cuda/equations/acoustic_lsrtm2d/kernels.cuh
src/sweep/csrc/cuda/equations/acoustic_lsrtm3d/acoustic_lsrtm3d.h
src/sweep/csrc/cuda/equations/acoustic_lsrtm3d/backward.cu
src/sweep/csrc/cuda/equations/acoustic_lsrtm3d/forward.cu
src/sweep/csrc/cuda/equations/acoustic_lsrtm3d/kernels.cu
src/sweep/csrc/cuda/equations/acoustic_lsrtm3d/kernels.cuh
src/sweep/csrc/cuda/equations/acoustic_vrz2d/acoustic_vrz2d.h
src/sweep/csrc/cuda/equations/acoustic_vrz2d/backward.cu
src/sweep/csrc/cuda/equations/acoustic_vrz2d/driver_traits.cuh
src/sweep/csrc/cuda/equations/acoustic_vrz2d/forward.cu
src/sweep/csrc/cuda/equations/acoustic_vrz2d/kernels.cuh
src/sweep/csrc/cuda/equations/acoustic_vrz3d/acoustic_vrz3d.h
src/sweep/csrc/cuda/equations/acoustic_vrz3d/backward.cu
src/sweep/csrc/cuda/equations/acoustic_vrz3d/forward.cu
src/sweep/csrc/cuda/equations/acoustic_vrz3d/kernels.cuh
src/sweep/csrc/cuda/equations/acoustic_vti_1st_2d/acoustic_vti_1st_2d.h
src/sweep/csrc/cuda/equations/acoustic_vti_1st_2d/backward.cu
src/sweep/csrc/cuda/equations/acoustic_vti_1st_2d/forward.cu
src/sweep/csrc/cuda/equations/acoustic_vti_1st_2d/kernels.cu
src/sweep/csrc/cuda/equations/acoustic_vti_1st_2d/kernels.cuh
src/sweep/csrc/cuda/equations/acoustic_vti_1st_3d/acoustic_vti_1st_3d.h
src/sweep/csrc/cuda/equations/acoustic_vti_1st_3d/backward.cu
src/sweep/csrc/cuda/equations/acoustic_vti_1st_3d/forward.cu
src/sweep/csrc/cuda/equations/acoustic_vti_1st_3d/kernels.cu
src/sweep/csrc/cuda/equations/acoustic_vti_1st_3d/kernels.cuh
src/sweep/csrc/cuda/equations/das2d/backward.cu
src/sweep/csrc/cuda/equations/das2d/das2d.h
src/sweep/csrc/cuda/equations/das2d/forward.cu
src/sweep/csrc/cuda/equations/das2d/kernels.cuh
src/sweep/csrc/cuda/equations/das3d/backward.cu
src/sweep/csrc/cuda/equations/das3d/das3d.h
src/sweep/csrc/cuda/equations/das3d/forward.cu
src/sweep/csrc/cuda/equations/das3d/kernels.cuh
src/sweep/csrc/cuda/equations/das_mu2d/backward.cu
src/sweep/csrc/cuda/equations/das_mu2d/das_mu2d.h
src/sweep/csrc/cuda/equations/das_mu2d/driver_traits.cuh
src/sweep/csrc/cuda/equations/das_mu2d/forward.cu
src/sweep/csrc/cuda/equations/das_mu2d/kernels.cuh
src/sweep/csrc/cuda/equations/das_mu3d/backward.cu
src/sweep/csrc/cuda/equations/das_mu3d/das_mu3d.h
src/sweep/csrc/cuda/equations/das_mu3d/driver_traits.cuh
src/sweep/csrc/cuda/equations/das_mu3d/forward.cu
src/sweep/csrc/cuda/equations/das_mu3d/kernels.cuh
src/sweep/csrc/cuda/equations/elastic2d/backward.cu
src/sweep/csrc/cuda/equations/elastic2d/driver_traits.cuh
src/sweep/csrc/cuda/equations/elastic2d/elastic2d.h
src/sweep/csrc/cuda/equations/elastic2d/forward.cu
src/sweep/csrc/cuda/equations/elastic2d/kernels.cu
src/sweep/csrc/cuda/equations/elastic2d/kernels.cuh
src/sweep/csrc/cuda/equations/elastic3d/backward.cu
src/sweep/csrc/cuda/equations/elastic3d/driver_traits.cuh
src/sweep/csrc/cuda/equations/elastic3d/elastic3d.h
src/sweep/csrc/cuda/equations/elastic3d/forward.cu
src/sweep/csrc/cuda/equations/elastic3d/kernels.cuh
src/sweep/csrc/cuda/equations/elastic_tti_2nd2d/backward.cu
src/sweep/csrc/cuda/equations/elastic_tti_2nd2d/elastic_tti_2nd2d.h
src/sweep/csrc/cuda/equations/elastic_tti_2nd2d/forward.cu
src/sweep/csrc/cuda/equations/elastic_tti_2nd2d/kernels.cuh
src/sweep/csrc/cuda/equations/elastic_tti_2nd2d/tensors.h
src/sweep/csrc/cuda/equations/elastic_tti_sg2d/backward.cu
src/sweep/csrc/cuda/equations/elastic_tti_sg2d/driver_traits.cuh
src/sweep/csrc/cuda/equations/elastic_tti_sg2d/elastic_tti_sg2d.h
src/sweep/csrc/cuda/equations/elastic_tti_sg2d/forward.cu
src/sweep/csrc/cuda/equations/elastic_tti_sg2d/kernels.cuh
src/sweep/csrc/cuda/equations/elastic_tti_sg2d/tensors.h
src/sweep/csrc/cuda/equations/elastic_tti_sg3d/backward.cu
src/sweep/csrc/cuda/equations/elastic_tti_sg3d/driver_traits.cuh
src/sweep/csrc/cuda/equations/elastic_tti_sg3d/elastic_tti_sg3d.h
src/sweep/csrc/cuda/equations/elastic_tti_sg3d/forward.cu
src/sweep/csrc/cuda/equations/elastic_tti_sg3d/kernels.cuh
src/sweep/csrc/cuda/equations/elastic_tti_sg3d/tensors.h
src/sweep/csrc/cuda/equations/elastic_vr2d/backward.cu
src/sweep/csrc/cuda/equations/elastic_vr2d/driver_traits.cuh
src/sweep/csrc/cuda/equations/elastic_vr2d/elastic_vr2d.h
src/sweep/csrc/cuda/equations/elastic_vr2d/forward.cu
src/sweep/csrc/cuda/equations/elastic_vr2d/kernels.cuh
src/sweep/csrc/cuda/equations/visco_acoustic2d/backward.cu
src/sweep/csrc/cuda/equations/visco_acoustic2d/fft.cu
src/sweep/csrc/cuda/equations/visco_acoustic2d/forward.cu
src/sweep/csrc/cuda/equations/visco_acoustic2d/kernels.cuh
src/sweep/csrc/cuda/equations/visco_acoustic2d/spectral_ops.cuh
src/sweep/csrc/cuda/equations/visco_acoustic2d/visco_acoustic2d.h
src/sweep/csrc/cuda/launch/config.h
src/sweep/csrc/cuda/operators/dim.cuh
src/sweep/csrc/cuda/operators/gradient.cuh
src/sweep/csrc/cuda/operators/laplace.cuh
src/sweep/csrc/cuda/operators/staggered.cuh
src/sweep/csrc/shared/boundary_session.h
src/sweep/csrc/shared/session_handle.h
src/sweep/csrc/shared/wavetypes.h
src/sweep/datasets/__init__.py
src/sweep/datasets/_benchmarks.py
src/sweep/datasets/_cache.py
src/sweep/datasets/_formats.py
src/sweep/datasets/cli.py
src/sweep/datasets/marmousi.py
src/sweep/datasets/overthrust_2d.py
src/sweep/datasets/registry.py
src/sweep/equations/__init__.py
src/sweep/equations/_anisotropy_utils.py
src/sweep/equations/_cpml.py
src/sweep/equations/_edges.py
src/sweep/equations/_elastic_step_core.py
src/sweep/equations/_free_surface.py
src/sweep/equations/_registry.py
src/sweep/equations/_topography.py
src/sweep/equations/acoustic.py
src/sweep/equations/acoustic1st.py
src/sweep/equations/acoustic3d.py
src/sweep/equations/acoustic_aniso.py
src/sweep/equations/acoustic_curvilinear.py
src/sweep/equations/acoustic_lsrtm.py
src/sweep/equations/acoustic_lsrtm3d.py
src/sweep/equations/acoustic_vrr.py
src/sweep/equations/acoustic_vrz.py
src/sweep/equations/acoustic_vti_1st.py
src/sweep/equations/base.py
src/sweep/equations/cuda_layout.py
src/sweep/equations/das.py
src/sweep/equations/elastic.py
src/sweep/equations/elastic3d.py
src/sweep/equations/elastic_apm.py
src/sweep/equations/elastic_curvilinear.py
src/sweep/equations/elastic_tti.py
src/sweep/equations/elastic_tti_2nd.py
src/sweep/equations/elastic_tti_sg.py
src/sweep/equations/elastic_tti_sg3d.py
src/sweep/equations/elastic_vrr.py
src/sweep/equations/fields.py
src/sweep/equations/pml.py
src/sweep/equations/qP_tariq.py
src/sweep/equations/qP_tti.py
src/sweep/equations/qP_vti.py
src/sweep/equations/slot_table.py
src/sweep/equations/utils.py
src/sweep/equations/visco_acoustic.py
src/sweep/memory/__init__.py
src/sweep/memory/shape.py
src/sweep/memory/torch.py
src/sweep/operators/__init__.py
src/sweep/operators/factory.py
src/sweep/operators/general.py
src/sweep/operators/jax.py
src/sweep/operators/rsg.py
src/sweep/operators/torch.py
src/sweep/parallel/__init__.py
src/sweep/parallel/_topology.py
src/sweep/parallel/dd_propagator.py
src/sweep/parallel/dd_spec.py
src/sweep/parallel/fast_halo.py
src/sweep/parallel/halo.py
src/sweep/parallel/mesh.py
src/sweep/parallel/padding.py
src/sweep/parallel/pml.py
src/sweep/parallel/routing.py
src/sweep/propagator/__init__.py
src/sweep/propagator/_bs_dispatch.py
src/sweep/propagator/_c.py
src/sweep/propagator/_c.pyi
src/sweep/propagator/_call_params.py
src/sweep/propagator/_eager_boundary_saving.py
src/sweep/propagator/_jax_boundary_saving.py
src/sweep/propagator/_ring_geometry.py
src/sweep/propagator/_stepped.py
src/sweep/propagator/_torch_eager.py
src/sweep/propagator/_torch_eager_custom_grad.py
src/sweep/propagator/base.py
src/sweep/propagator/jax.py
src/sweep/propagator/jax.pyi
src/sweep/propagator/options.py
src/sweep/propagator/options.pyi
src/sweep/propagator/torch.py
src/sweep/propagator/torch.pyi
src/sweep/receivers/__init__.py
src/sweep/receivers/base.py
src/sweep/receivers/jax.py
src/sweep/receivers/torch.py
src/sweep/sources/__init__.py
src/sweep/sources/base.py
src/sweep/sources/jax.py
src/sweep/sources/torch.py
src/sweep/utils/__init__.py
src/sweep/utils/curvilinear.py
src/sweep/utils/general.py
src/sweep/utils/jax.py
src/sweep/utils/torch.py
src/sweep_solver.egg-info/PKG-INFO
src/sweep_solver.egg-info/SOURCES.txt
src/sweep_solver.egg-info/dependency_links.txt
src/sweep_solver.egg-info/entry_points.txt
src/sweep_solver.egg-info/requires.txt
src/sweep_solver.egg-info/top_level.txt
test/test_acoustic_curvilinear.py
test/test_acoustic_tti.py
test/test_acoustic_vti_1st.py
test/test_acoustic_vti_1st_c_adjoint.py
test/test_adcig_from_pool.py
test/test_aniso_free_surface_guard.py
test/test_aniso_pseudoacoustic_gradient.py
test/test_apm_backward_guard.py
test/test_apm_model_assembly.py
test/test_backend_c.py
test/test_backward_is_repeatable.py
test/test_backward_scratch_from_pool.py
test/test_balanced_grid.py
test/test_binding_is_mandatory.py
test/test_binding_names_resolve_from_c_name.py
test/test_body_force_rho_grad.py
test/test_boundary_compact_kernel.py
test/test_boundary_disk_staging_slot.py
test/test_boundary_disk_write_failure.py
test/test_boundary_dtype_env_leak.py
test/test_boundary_fp16_dynamic_range.py
test/test_boundary_int_idx_overflow.py
test/test_boundary_saving_topography_guard.py
test/test_boundary_spec_resolution.py
test/test_boundary_storage_dtype_validation.py
test/test_boundary_tail_truncation.py
test/test_bs_backward_last_two_alloc.py
test/test_bs_backward_memory.py
test/test_bs_reconstruction_from_pool.py
test/test_bs_strip_source.py
test/test_buf_views.py
test/test_c_aux_slab_repeatability.py
test/test_c_default_boundary_saving.py
test/test_checkpoint_batch_shrink.py
test/test_checkpoint_replay_from_pool.py
test/test_checkpoint_state_from_pool.py
test/test_cli_show.py
test/test_companion_aliases.py
test/test_compiled_call_params.py
test/test_compiled_del_no_setattr_typeerror.py
test/test_construction_only_attrs.py
test/test_core_arguments.py
test/test_core_capi.py
test/test_core_stream_scope.py
test/test_cpml_axis_update.py
test/test_cuda_options_typed_memory.py
test/test_custom_gradient.py
test/test_cut_face_mask.py
test/test_das3d_boundary_capability.py
test/test_das_equations.py
test/test_datasets_registry.py
test/test_dd_backward_two_tile.py
test/test_dd_backward_two_tile_3d.py
test/test_dd_elastic_backward_two_tile.py
test/test_dd_elastic_tiles_3d.py
test/test_dd_elastic_two_tile.py
test/test_dd_gather_record_shot_groups.py
test/test_dd_memory_strategy_admission.py
test/test_dd_pml_type_inherited.py
test/test_dd_refuses_multi_shot_geometry.py
test/test_dd_supported_equations.py
test/test_dd_tail_two_tile.py
test/test_dd_tile_memory_strategy.py
test/test_dd_tiles_3d.py
test/test_dd_topography_admission.py
test/test_dd_two_tile.py
test/test_decimate_returns_a_copy.py
test/test_derived_models_from_pool.py
test/test_dynamo_recompile_limit.py
test/test_eager_boundary_saving.py
test/test_eager_only_stub_equations.py
test/test_eager_record_is_stacked.py
test/test_eager_snapshot_buffer.py
test/test_elastic_apm.py
test/test_elastic_batched_models.py
test/test_elastic_c_adjoint_receiver.py
test/test_elastic_curvilinear.py
test/test_elastic_tti.py
test/test_elastic_tti_2nd.py
test/test_elastic_tti_sg3d.py
test/test_elastic_vector_reflectivity.py
test/test_elastic_vr_cuda_parity.py
test/test_elastic_vr_gradient_consistency.py
test/test_equation_registry_public_api.py
test/test_equations.py
test/test_forward_scratch_from_pool.py
test/test_free_surface_concat.py
test/test_free_surface_per_edge.py
test/test_free_surface_per_edge_cuda.py
test/test_free_surface_topo.py
test/test_gradients_from_pool.py
test/test_history_from_pool.py
test/test_illumination_pin.py
test/test_image_topography_capability.py
test/test_impl_demotion_keeps_memory.py
test/test_impl_resolution.py
test/test_input_core_generated.py
test/test_installation_smoke.py
test/test_jax_boundary_saving.py
test/test_jax_jit_tracing.py
test/test_jit_can_compile_without_gpu.py
test/test_jit_stage_staleness.py
test/test_kwarg_validation.py
test/test_memory_api_resolution.py
test/test_memory_capabilities.py
test/test_memory_legacy_spellings.py
test/test_memory_strategy_select.py
test/test_memory_strategy_types.py
test/test_model_parallel_halo.py
test/test_model_parallel_padding.py
test/test_model_parallel_pml.py
test/test_model_parallel_topology.py
test/test_multi_device_cuda_guard.py
test/test_no_private_aten_headers.py
test/test_op.py
test/test_pml_cache_follows_the_equation.py
test/test_prebuilt_binding_available.py
test/test_prebuilt_install_path.py
test/test_propagator_exports.py
test/test_propjax_call_api.py
test/test_record_from_pool.py
test/test_requires_binding.py
test/test_rsg_stencil_jax.py
test/test_segy_block_read.py
test/test_shipped_core.py
test/test_slot_table_consistency.py
test/test_source_io_contract.py
test/test_source_mask_only_when_read.py
test/test_stepped_backward.py
test/test_stepped_backward_elastic.py
test/test_stepped_forward.py
test/test_stepped_forward_elastic.py
test/test_sweep_pytorch.py
test/test_topography_acoustic2d.py
test/test_topography_elastic2d.py
test/test_topography_helpers.py
test/test_topography_runtime_build.py
test/test_type_stubs_match_runtime.py
test/test_use_ckpt_is_read_only.py
test/test_visco_acoustic.py
test/test_visco_acoustic_cuda.py
test/test_visco_acoustic_jax.py
test/test_vrz_boundary_imaging.py
test/test_vrz_exact_cpml_adjoint.py
test/test_vti1st_anisotropic_spacing.py
test/test_workspace_is_lazy.py