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/_jit.py
src/sweep/cli.py
src/sweep/scalars.py
src/sweep/signal.py
src/sweep/backend/__init__.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/csrc/CMakeLists.txt
src/sweep/csrc/bindings/bindings_utils.h
src/sweep/csrc/bindings/module.cpp
src/sweep/csrc/cpu/cpu_binding.cpp
src/sweep/csrc/cpu/cpu_binding.h
src/sweep/csrc/cpu/cpu_binding_stub.cpp
src/sweep/csrc/cpu/common/cpu_engine.cpp
src/sweep/csrc/cpu/common/cpu_engine.h
src/sweep/csrc/cpu/equations/acoustic2d/acoustic2d_cpu.cpp
src/sweep/csrc/cpu/equations/acoustic2d/acoustic2d_cpu.h
src/sweep/csrc/cpu/equations/acoustic3d/acoustic3d_cpu.cpp
src/sweep/csrc/cpu/equations/acoustic3d/acoustic3d_cpu.h
src/sweep/csrc/cpu/equations/acoustic_lsrtm2d/acoustic_lsrtm2d_cpu.cpp
src/sweep/csrc/cpu/equations/acoustic_lsrtm2d/acoustic_lsrtm2d_cpu.h
src/sweep/csrc/cpu/equations/acoustic_lsrtm3d/acoustic_lsrtm3d_cpu.cpp
src/sweep/csrc/cpu/equations/acoustic_lsrtm3d/acoustic_lsrtm3d_cpu.h
src/sweep/csrc/cpu/equations/acoustic_vrz2d/acoustic_vrz2d_cpu.cpp
src/sweep/csrc/cpu/equations/acoustic_vrz2d/acoustic_vrz2d_cpu.h
src/sweep/csrc/cpu/equations/acoustic_vrz3d/acoustic_vrz3d_cpu.cpp
src/sweep/csrc/cpu/equations/acoustic_vrz3d/acoustic_vrz3d_cpu.h
src/sweep/csrc/cpu/equations/das2d/das2d_cpu.cpp
src/sweep/csrc/cpu/equations/das2d/das2d_cpu.h
src/sweep/csrc/cpu/equations/das3d/das3d_cpu.cpp
src/sweep/csrc/cpu/equations/das3d/das3d_cpu.h
src/sweep/csrc/cpu/equations/das_mu2d/das_mu2d_cpu.cpp
src/sweep/csrc/cpu/equations/das_mu2d/das_mu2d_cpu.h
src/sweep/csrc/cpu/equations/das_mu3d/das_mu3d_cpu.cpp
src/sweep/csrc/cpu/equations/das_mu3d/das_mu3d_cpu.h
src/sweep/csrc/cpu/equations/elastic2d/elastic2d_cpu.cpp
src/sweep/csrc/cpu/equations/elastic2d/elastic2d_cpu.h
src/sweep/csrc/cpu/equations/elastic3d/elastic3d_cpu.cpp
src/sweep/csrc/cpu/equations/elastic3d/elastic3d_cpu.h
src/sweep/csrc/cpu/equations/elastic_tti_sg2d/elastic_tti_sg2d_cpu.cpp
src/sweep/csrc/cpu/equations/elastic_tti_sg2d/elastic_tti_sg2d_cpu.h
src/sweep/csrc/cpu/operators/fd.h
src/sweep/csrc/cuda/common/acoustic.h
src/sweep/csrc/cuda/common/acoustic_vrz_fused.cuh
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/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/elastic.h
src/sweep/csrc/cuda/common/elastic_free_surface.cuh
src/sweep/csrc/cuda/common/wavetypes.h
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/types.cuh
src/sweep/csrc/cuda/equations/acoustic2d/acoustic2d.h
src/sweep/csrc/cuda/equations/acoustic2d/backward.cu
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/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/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/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/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/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/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_sg2d/backward.cu
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_vr2d/backward.cu
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/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/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/_elastic_step_core.py
src/sweep/equations/_free_surface.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/aec.py
src/sweep/equations/aec_lsrtm.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/elasticP.py
src/sweep/equations/elastic_apm.py
src/sweep/equations/elastic_curvilinear.py
src/sweep/equations/elastic_lsrtm.py
src/sweep/equations/elastic_tti.py
src/sweep/equations/elastic_tti_sg.py
src/sweep/equations/elastic_vrr.py
src/sweep/equations/elasticz.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/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/propagator/__init__.py
src/sweep/propagator/_bs_dispatch.py
src/sweep/propagator/_c.py
src/sweep/propagator/_c.pyi
src/sweep/propagator/_eager_boundary_saving.py
src/sweep/propagator/_jax_boundary_saving.py
src/sweep/propagator/_ring_geometry.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_boundary_dtype_env_leak.py
test/test_boundary_int_idx_overflow.py
test/test_boundary_storage_dtype_validation.py
test/test_c_default_boundary_saving.py
test/test_checkpoint_batch_shrink.py
test/test_companion_aliases.py
test/test_compiled_del_no_setattr_typeerror.py
test/test_custom_gradient.py
test/test_das_equations.py
test/test_datasets_registry.py
test/test_eager_boundary_saving.py
test/test_elastic_apm.py
test/test_elastic_batched_models.py
test/test_elastic_curvilinear.py
test/test_elastic_tti.py
test/test_elastic_vector_reflectivity.py
test/test_elastic_vr_cuda_parity.py
test/test_elastic_vr_gradient_consistency.py
test/test_equations.py
test/test_free_surface_concat.py
test/test_free_surface_topo.py
test/test_installation_smoke.py
test/test_jax_boundary_saving.py
test/test_jax_jit_tracing.py
test/test_multi_device_cuda_guard.py
test/test_op.py
test/test_source_io_contract.py
test/test_sweep_pytorch.py
test/test_topography_acoustic2d.py
test/test_topography_elastic2d.py
test/test_topography_helpers.py
test/test_visco_acoustic.py
test/test_visco_acoustic_jax.py