Skip to main content

edmkit-search

Trajectory construction on an energy landscape, packaged as a library on top of edmkit.

Install

pip install edmkit-search
# or
uv add edmkit-search

Overview

The search loop builds a trajectory step by step:

  1. From the current frontier (batch of (state, context)),
  2. expand neighbors via a Neighborhood,
  3. score children with an Energy to obtain (energies, contexts),
  4. and pick the next frontier with a Strategy.

See CODING.md for the design rules. The subpackages mirror the four abstractions: energy/, neighborhood/, state/, strategy/, plus dataset/ for the input containers.

Quick start

uv sync
PYTHON_GIL=0 uv run python e2e/synthetic.py

e2e/synthetic.py generates a Lorenz-96 trajectory mixed with noise columns and uses greedy forward selection to recover the informative subset.

Tests

uv run pytest

Metadata

Release files for edmkit-search 0.2.1

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for edmkit-search 0.2.1
File Size Uploaded
edmkit_search-0.2.1.tar.gz 12.3 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for edmkit-search 0.2.1
File Interpreter ABI Platform
edmkit_search-0.2.1-py3-none-any.whl Python 3 none any Details

Total release size: 37.2 kB

Release files / edmkit_search-0.2.1.tar.gz

Download URL edmkit_search-0.2.1.tar.gz
Size 12.3 kB
Tags Source
SHA-256 checksum
How to use checksums
a4e9cf0026da44eabecb3fc28478004d58d59b0c33e522937e37527dc51d716e
BLAKE2b-256 checksum
How to use checksums
60ddebdb786c2c728cd27385e4558612871a9716a526a4abc6edab01b2095ca5
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.12.3 {"installer":{"name":"uv","version":"0.12.3","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / edmkit_search-0.2.1-py3-none-any.whl

Download URL edmkit_search-0.2.1-py3-none-any.whl
Size 24.9 kB
Tags Python 3
SHA-256 checksum
How to use checksums
149cc9eb86bed9748a4f4d1128b02446613ba1aae90f6d168c42f447b963ebc8
BLAKE2b-256 checksum
How to use checksums
4259dafb5a8ce20b49789bbd9224569a58fcbd5ceead173f2f0aa8a5c242adff
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.12.3 {"installer":{"name":"uv","version":"0.12.3","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page