PACKAGE · import sweep

sweep-solver

The differentiable wave-equation engine. Nine equation families — acoustic, elastic, VTI, TTI, VRZ, SEM — on torch / JAX / native-CUDA backends, with FWI, LSRTM and RTM building blocks.

Open the solver docs →

PACKAGE · import sweep_agent

sweep-agent

The natural-language control layer. Say what you want and a local LLM (Ollama / vLLM) turns it into a validated sweep run — forward modelling, benchmark models, plots.

Open the agent docs →

COMPANIONS · GROWING

…and the family

sweep-tasks (production FWI/LSRTM runner), sweep-loss, sweep-nn, sweep-tomo — each an independent package, added to sweepx as it ships.

Browse the org →

ONE INSTALL · MANY PACKAGES

One install.
Then import sweep.

sweepx is the umbrella — it carries no code of its own, it just pulls in the engine and its published companions. Install sweepx, but import sweep (same pattern as scikit-learn → sklearn).

# one install — pulls the engine + the agent:
#     pip install sweepx

import sweep                                       # ← sweep-solver, the engine
from sweep.propagator.torch import PropTorch       # torch / JAX / native-CUDA
from sweep.equations import Acoustic, ElasticTTI   # nine equation families

# …or drive it all in plain language — sweep-agent:
#     sweep-agent chat
#     >>> load Marmousi and run a forward — show the shot gather

Install & get started →

THE FAMILY

Every piece, installable on its own.

The same layout as the PyLops family — sweepx bundles the published packages; you can also pip install any single one.

PUBLISHED

sweep-solver

import sweep

Wave-equation engine — equations, propagators, FWI/LSRTM/RTM.

PUBLISHED

sweep-agent

import sweep_agent

Natural-language control via a local LLM.

COMPANION

sweep-tasks

coming

Production FWI/LSRTM runner — specs, YAML, multi-GPU, IO.

COMPANION

sweep-loss

coming

Misfit / loss functions for inversion.

COMPANION

sweep-nn

coming

Neural reparameterizations — INR / hash / SIREN encoders.

COMPANION

sweep-tomo

coming

First-arrival traveltime tomography — eikonal + SIRT / FATT.