genforge 0.1.0__tar.gz

genforge-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Monish Lokhande
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

genforge-0.1.0/PKG-INFO

@@ -0,0 +1,124 @@
+Metadata-Version: 2.4
+Name: genforge
+Version: 0.1.0
+Summary: A unified, PyTorch-based framework for generative-modeling techniques with a clean control layer.
+Keywords: generative-models,diffusion,flow-matching,score-based,discrete-diffusion,guidance,control,pytorch
+Author: Monish Lokhande
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Dist: torch>=2.2
+Requires-Dist: numpy>=1.26
+Requires-Dist: hydra-core>=1.3
+Requires-Dist: omegaconf>=2.3
+Requires-Dist: scipy>=1.11 ; extra == 'flow'
+Requires-Dist: wandb>=0.16 ; extra == 'logging'
+Requires-Dist: tqdm>=4.66 ; extra == 'logging'
+Requires-Dist: tiktoken>=0.7 ; extra == 'text'
+Requires-Dist: datasets>=2.18 ; extra == 'text'
+Requires-Python: >=3.11
+Project-URL: Homepage, https://github.com/MonishLokhande/genforge
+Project-URL: Repository, https://github.com/MonishLokhande/genforge
+Project-URL: Documentation, https://github.com/MonishLokhande/genforge#readme
+Project-URL: Issues, https://github.com/MonishLokhande/genforge/issues
+Provides-Extra: flow
+Provides-Extra: logging
+Provides-Extra: text
+Description-Content-Type: text/markdown
+
+# genforge
+
+A unified, PyTorch framework for generative modeling — score/diffusion SDEs, probability-flow
+ODEs, flow matching / OT-CFM, stochastic interpolants, and discrete (D3PM-style) diffusion — with a
+clean **control layer** for conditioning, guidance, constraints, and amortized control.
+
+The idea: a generative model is one process that turns a simple starting distribution (usually noise)
+into data, and every way of steering it —
+conditioning, guidance, constraints, planning — is the **same move**: reweight that process so the
+outcomes you want become more likely. The process, the steering, and how the steering is approximated
+are separate, swappable parts. A full documentation site (architecture, experiments, extending) is coming soon.
+
+## Install
+
+**Use it as a library:**
+
+```bash
+pip install genforge          # or: uv add genforge
+```
+
+The bundled `experiment/` tree used in the Quickstart below ships with the **source clone**, not
+the PyPI wheel — running `forge train experiment=...` requires `git clone` + `uv sync`, not just
+`pip install genforge`.
+
+**Develop / run the bundled experiments:**
+
+```bash
+git clone https://github.com/MonishLokhande/genforge
+cd genforge
+uv sync                   # core (light: 2-D distributions)
+uv sync --extra flow      # + OT-CFM (scipy)
+uv sync --extra text      # + real BPE / streamed corpora (tiktoken, datasets)
+uv sync --extra logging   # + experiment logging (wandb) + progress bars (tqdm)
+```
+
+Robotics adapters are a **dependency group**, not an extra (one dependency installs from git):
+`uv sync --group robotics` (mujoco, robomimic, gym-pusht/aloha, minari).
+
+## Quickstart
+
+```bash
+uv run forge list                                       # registered components
+uv run forge train  experiment=distributions/ddpm/base
+uv run forge sample experiment=distributions/ddpm/base  # or: sample checkpoint=<path>.pt
+```
+
+## Layout
+
+Every component registers via `@register(category, name)` and is wired by a config-driven builder in
+dependency order — adding one is a single decorated class plus a config leaf, no other wiring.
+
+- **`src/forge/`** — the framework only: `core` (registry · builder · interfaces · protocols ·
+  plugins), `spaces`, `schedules`, `models`, `methods`, `samplers`, `costs`, `control`,
+  `preprocessing`, `runners`. Protocols, ABCs, and generic utilities — never concrete env code.
+- **`envs/`** — concrete, swappable **data-source plugins** (environment + dataset + processor per
+  package). An experiment loads them via its `plugins:` field; they are not baked into the core.
+  Contracts: [`core/protocols.py`](src/forge/core/protocols.py).
+- **`experiment/`** — Hydra base+delta bundles, selected with `experiment=<family>/<variant>/<method>`.
+
+## Experiments
+
+Selected with `experiment=<family>/<variant>/<method>`.
+
+| Family | What |
+|---|---|
+| `distributions/*` | Continuous 2-D — DDPM, flow matching, stochastic-interpolant SDE, DDIM; control via projection / guidance / CBF; value guidance. |
+| `discrete/d3pm/base` | Discrete (absorbing) diffusion on a toy categorical target. |
+| `text/char/*` | Discrete diffusion LM, char-level — `d3pm` / `mdlm` / `sedd`. |
+| `text/tinystories/*` | The **same** methods at real GPT-2 BPE (vocab 50258, needs `--extra text`). |
+| `trajectory/plan/base` | Goal-conditioned trajectory planning (flat-tensor windowing, endpoint-pinned). |
+| `robotics/*` | Offline-RL trajectory planning (maze2d, locomotion) and closed-loop diffusion policies (robomimic, pusht, aloha); needs `--group robotics`. |
+
+`text/char/*` and `text/tinystories/*` are two variants of one **`text` family** (a single env plugin
+`envs.text` registers both) — the same absorbing + transformer + `{d3pm,mdlm,sedd}` rig; only the
+tokenizer (char vs. real BPE) and scale differ.
+
+```bash
+uv run forge train experiment=text/char/d3pm/small                     # char-level LM
+uv run --extra text forge train experiment=text/tinystories/d3pm/small # same method, real BPE
+```
+
+## Documentation
+
+A full documentation site — installation, architecture, experiments, and extending — is coming soon.
+Until then, this README plus the inline docstrings across `src/forge/` are the reference.
+
+## License
+
+MIT — see [LICENSE](LICENSE).
+
+## Citation
+
+If you use genforge in your work, please cite it — citation metadata is in [CITATION.cff](CITATION.cff).

genforge-0.1.0/README.md

@@ -0,0 +1,93 @@
+# genforge
+
+A unified, PyTorch framework for generative modeling — score/diffusion SDEs, probability-flow
+ODEs, flow matching / OT-CFM, stochastic interpolants, and discrete (D3PM-style) diffusion — with a
+clean **control layer** for conditioning, guidance, constraints, and amortized control.
+
+The idea: a generative model is one process that turns a simple starting distribution (usually noise)
+into data, and every way of steering it —
+conditioning, guidance, constraints, planning — is the **same move**: reweight that process so the
+outcomes you want become more likely. The process, the steering, and how the steering is approximated
+are separate, swappable parts. A full documentation site (architecture, experiments, extending) is coming soon.
+
+## Install
+
+**Use it as a library:**
+
+```bash
+pip install genforge          # or: uv add genforge
+```
+
+The bundled `experiment/` tree used in the Quickstart below ships with the **source clone**, not
+the PyPI wheel — running `forge train experiment=...` requires `git clone` + `uv sync`, not just
+`pip install genforge`.
+
+**Develop / run the bundled experiments:**
+
+```bash
+git clone https://github.com/MonishLokhande/genforge
+cd genforge
+uv sync                   # core (light: 2-D distributions)
+uv sync --extra flow      # + OT-CFM (scipy)
+uv sync --extra text      # + real BPE / streamed corpora (tiktoken, datasets)
+uv sync --extra logging   # + experiment logging (wandb) + progress bars (tqdm)
+```
+
+Robotics adapters are a **dependency group**, not an extra (one dependency installs from git):
+`uv sync --group robotics` (mujoco, robomimic, gym-pusht/aloha, minari).
+
+## Quickstart
+
+```bash
+uv run forge list                                       # registered components
+uv run forge train  experiment=distributions/ddpm/base
+uv run forge sample experiment=distributions/ddpm/base  # or: sample checkpoint=<path>.pt
+```
+
+## Layout
+
+Every component registers via `@register(category, name)` and is wired by a config-driven builder in
+dependency order — adding one is a single decorated class plus a config leaf, no other wiring.
+
+- **`src/forge/`** — the framework only: `core` (registry · builder · interfaces · protocols ·
+  plugins), `spaces`, `schedules`, `models`, `methods`, `samplers`, `costs`, `control`,
+  `preprocessing`, `runners`. Protocols, ABCs, and generic utilities — never concrete env code.
+- **`envs/`** — concrete, swappable **data-source plugins** (environment + dataset + processor per
+  package). An experiment loads them via its `plugins:` field; they are not baked into the core.
+  Contracts: [`core/protocols.py`](src/forge/core/protocols.py).
+- **`experiment/`** — Hydra base+delta bundles, selected with `experiment=<family>/<variant>/<method>`.
+
+## Experiments
+
+Selected with `experiment=<family>/<variant>/<method>`.
+
+| Family | What |
+|---|---|
+| `distributions/*` | Continuous 2-D — DDPM, flow matching, stochastic-interpolant SDE, DDIM; control via projection / guidance / CBF; value guidance. |
+| `discrete/d3pm/base` | Discrete (absorbing) diffusion on a toy categorical target. |
+| `text/char/*` | Discrete diffusion LM, char-level — `d3pm` / `mdlm` / `sedd`. |
+| `text/tinystories/*` | The **same** methods at real GPT-2 BPE (vocab 50258, needs `--extra text`). |
+| `trajectory/plan/base` | Goal-conditioned trajectory planning (flat-tensor windowing, endpoint-pinned). |
+| `robotics/*` | Offline-RL trajectory planning (maze2d, locomotion) and closed-loop diffusion policies (robomimic, pusht, aloha); needs `--group robotics`. |
+
+`text/char/*` and `text/tinystories/*` are two variants of one **`text` family** (a single env plugin
+`envs.text` registers both) — the same absorbing + transformer + `{d3pm,mdlm,sedd}` rig; only the
+tokenizer (char vs. real BPE) and scale differ.
+
+```bash
+uv run forge train experiment=text/char/d3pm/small                     # char-level LM
+uv run --extra text forge train experiment=text/tinystories/d3pm/small # same method, real BPE
+```
+
+## Documentation
+
+A full documentation site — installation, architecture, experiments, and extending — is coming soon.
+Until then, this README plus the inline docstrings across `src/forge/` are the reference.
+
+## License
+
+MIT — see [LICENSE](LICENSE).
+
+## Citation
+
+If you use genforge in your work, please cite it — citation metadata is in [CITATION.cff](CITATION.cff).

genforge-0.1.0/pyproject.toml

@@ -0,0 +1,96 @@
+[project]
+name = "genforge"
+version = "0.1.0"
+description = "A unified, PyTorch-based framework for generative-modeling techniques with a clean control layer."
+readme = "README.md"
+license = "MIT"
+license-files = ["LICENSE"]
+authors = [
+    { name = "Monish Lokhande" }
+]
+requires-python = ">=3.11"
+keywords = [
+    "generative-models", "diffusion", "flow-matching", "score-based",
+    "discrete-diffusion", "guidance", "control", "pytorch",
+]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Science/Research",
+    "Programming Language :: Python :: 3.11",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+]
+dependencies = [
+    "torch>=2.2",
+    "numpy>=1.26",
+    "hydra-core>=1.3",
+    "omegaconf>=2.3",
+]
+
+[project.optional-dependencies]
+# Optional extras keep the 2-D distributions core light.
+flow = ["scipy>=1.11"]   # minibatch-OT coupling for ot_cfm
+text = ["tiktoken>=0.7", "datasets>=2.18"]   # real GPT-2 BPE + streamed corpora (e.g. TinyStories)
+logging = ["wandb>=0.16", "tqdm>=4.66"]   # optional experiment logging + progress bars (off by default)
+
+[project.urls]
+Homepage = "https://github.com/MonishLokhande/genforge"
+Repository = "https://github.com/MonishLokhande/genforge"
+Documentation = "https://github.com/MonishLokhande/genforge#readme"
+Issues = "https://github.com/MonishLokhande/genforge/issues"
+
+[dependency-groups]
+dev = [
+    "pytest>=8.0",
+]
+docs = [
+    "mkdocs-material>=9.7.6",
+]
+# Offline-RL / robot adapters (maze2d / locomotion / hf_lowdim / robomimic). Pins copied from an offline-RL reference implementation.
+# A dependency-group, not an extra: it contains a git direct-reference (robomimic) that PyPI rejects
+# in published wheel metadata. Sync locally with: uv sync --group robotics
+robotics = [
+    "h5py>=3.16",                # d4rl / robomimic HDF5 adapters read buffers directly
+    "gym-pusht>=0.1.6",
+    "gym-aloha>=0.1.3",
+    "gymnasium>=1.3.0",
+    "gymnasium-robotics>=1.4.2",
+    "datasets>=2.19.0",
+    "minari[hdf5,hf]>=0.5.3",
+    "mujoco>=3.8.0",
+    "robomimic @ git+https://github.com/ARISE-Initiative/robomimic.git@v0.5.0",
+    "robosuite>=1.5",           # unpinned backtracks to 0.3.0 -> deprecated mujoco-py; >=1.5 avoids it
+]
+
+[tool.uv]
+constraint-dependencies = ["pymunk<7", "numba>=0.65"]  # keep robosuite off the py<3.10 numba 0.53
+# robomimic 0.5 hard-pins an old HF stack; relax to >= so the modern stack is kept.
+override-dependencies = [
+    "transformers>=4.41.2",
+    "diffusers>=0.11.1",
+    "huggingface-hub>=0.23.4",
+]
+
+[tool.uv.build-backend]
+module-name = "forge"
+
+[[tool.uv.index]]
+name = "testpypi"
+url = "https://test.pypi.org/simple/"
+publish-url = "https://test.pypi.org/legacy/"
+default = false
+# Publish-only: keep this index out of dependency resolution so a stray package on TestPyPI
+# (e.g. an old hydra-core) can't shadow the real one on PyPI. `uv publish --index testpypi`
+# still works via publish-url. Scope any TestPyPI-sourced dep via [tool.uv.sources] instead.
+explicit = true
+
+[tool.pytest.ini_options]
+# Put the repo root on sys.path for the test session so the repo-root `envs/*` plugin packages are
+# importable (the package install only adds `src/` via its .pth). conftest.py then registers them.
+pythonpath = ["."]
+
+[project.scripts]
+forge = "forge.cli:main"
+
+[build-system]
+requires = ["uv_build>=0.11.7,<0.12.0"]
+build-backend = "uv_build"

genforge-0.1.0/src/forge/__init__.py

@@ -0,0 +1,3 @@
+"""genforge — a unified, PyTorch-based framework for generative-modeling techniques."""
+
+__version__ = "0.1.0"

genforge-0.1.0/src/forge/cli.py

@@ -0,0 +1,143 @@
+"""The ``forge`` command-line entrypoint: ``list`` / ``train`` / ``sample``.
+
+``list`` imports the built-ins so registrations fire, then prints the registered components by
+category. ``train`` / ``sample`` compose a Hydra config from an ``experiment=`` selection, build the
+runner, and run. ``sample checkpoint=<path.pt>`` rebuilds everything from the self-contained
+checkpoint alone (Invariant 5). All three fail loudly on misconfiguration.
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from typing import Optional, Sequence
+
+from .core import registry
+from .core.builder import build, import_builtin_components
+
+_EXPERIMENT_HINT = (
+    "Select one with `experiment=<env>/<params>/<method>` "
+    "(e.g. `forge train experiment=distributions/ddpm/base`)."
+)
+
+
+def _split_overrides(overrides: Sequence[str]) -> dict:
+    """Parse ``key=value`` overrides into a flat dict (first '=')."""
+    flat: dict[str, str] = {}
+    for o in overrides:
+        if "=" in o:
+            k, v = o.split("=", 1)
+            flat[k] = v
+    return flat
+
+
+def _cmd_list(_args: argparse.Namespace) -> int:
+    import_builtin_components()
+    # Concrete envs are plugins (no experiment selected here), so import the bundled env packages
+    # too — otherwise `list` would omit environments/datasets/env-preprocessors.
+    from .core.plugins import load_bundled_envs
+
+    load_bundled_envs()
+    reg = registry.registered()
+    print("forge components")
+    print("===================")
+    for category in registry.CATEGORIES:
+        comps = reg.get(category, {})
+        names = ", ".join(comps) if comps else "(none yet)"
+        print(f"  {category:<13} {names}")
+    for category in [c for c in reg if c not in registry.CATEGORIES]:
+        print(f"  {category:<13} {', '.join(reg[category])}")
+    return 0
+
+
+def _run_from_config(overrides: Sequence[str], action: str) -> int:
+    from omegaconf import OmegaConf
+
+    from .core.compose import compose_config
+
+    cfg = compose_config(overrides)
+    runner = build(cfg)
+    runner.resolved_config = OmegaConf.to_container(cfg, resolve=True)
+
+    if action == "train":
+        runner.train()
+        metrics = runner.evaluate()
+        print(f"[train] done. eval: {metrics}")
+        return 0
+
+    # sample from an experiment: load its configured checkpoint if present.
+    ckpt_path = getattr(runner, "ckpt_path", None)
+    if ckpt_path:
+        from pathlib import Path
+
+        from .core.checkpoint import load_checkpoint
+
+        if not Path(ckpt_path).exists():
+            print(
+                f"`forge sample` found no checkpoint at {ckpt_path!r}. Train first "
+                f"(`forge train experiment=...`) or pass `checkpoint=<path.pt>`.",
+                file=sys.stderr,
+            )
+            return 1
+        runner.load_state(load_checkpoint(ckpt_path))
+    metrics = runner.evaluate()
+    print(f"[sample] {metrics}")
+    return 0
+
+
+def _cmd_train(args: argparse.Namespace) -> int:
+    flat = _split_overrides(args.overrides)
+    if "experiment" not in flat:
+        print(f"`forge train` requires an experiment selection. {_EXPERIMENT_HINT}", file=sys.stderr)
+        return 2
+    return _run_from_config(args.overrides, "train")
+
+
+def _cmd_sample(args: argparse.Namespace) -> int:
+    flat = _split_overrides(args.overrides)
+    if "checkpoint" in flat:
+        # Self-contained path: rebuild from the .pt alone (Invariant 5).
+        from .runners.training import TrainingRunner
+
+        runner = TrainingRunner.from_checkpoint(flat["checkpoint"], build_fn=build)
+        metrics = runner.evaluate()
+        print(f"[sample] from checkpoint {flat['checkpoint']}: {metrics}")
+        return 0
+    if "experiment" not in flat:
+        print(
+            f"`forge sample` requires `experiment=...` or `checkpoint=<path.pt>`. {_EXPERIMENT_HINT}",
+            file=sys.stderr,
+        )
+        return 2
+    return _run_from_config(args.overrides, "sample")
+
+
+def _build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        prog="forge",
+        description="A unified framework for generative modeling with a clean control layer.",
+    )
+    sub = parser.add_subparsers(dest="command", required=True)
+
+    p_list = sub.add_parser("list", help="List registered components by category.")
+    p_list.set_defaults(func=_cmd_list)
+
+    p_train = sub.add_parser("train", help="Train a model from an experiment config.")
+    p_train.add_argument("overrides", nargs="*", help="Hydra-style overrides, e.g. experiment=...")
+    p_train.set_defaults(func=_cmd_train)
+
+    p_sample = sub.add_parser("sample", help="Sample from a trained model or checkpoint.")
+    p_sample.add_argument("overrides", nargs="*", help="experiment=... or checkpoint=<path.pt>")
+    p_sample.set_defaults(func=_cmd_sample)
+
+    return parser
+
+
+def main(argv: Optional[Sequence[str]] = None) -> int:
+    parser = _build_parser()
+    args = parser.parse_args(argv)
+    return args.func(args)
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

genforge-0.1.0/src/forge/configs/config.yaml

@@ -0,0 +1,22 @@
+# genforge root config (Hydra base+delta).
+#
+# Per-category config groups live in the sibling directories (space/, schedule/, model/, ...).
+# Experiments are base+delta bundles in the repo-root `experiment/` tree, which is added to the
+# Hydra searchpath at runtime by the CLI (env var GENFORGE_EXP_ROOT). Select an experiment with
+# `experiment=<env>/<params>/<method>` — it composes the component groups it needs.
+
+defaults:
+  - _self_
+  - experiment: ???        # mandatory: train/sample require an experiment selection
+
+# Global run settings (resolved into the checkpoint, Invariant 5).
+seed: 0
+
+hydra:
+  searchpath:
+    - file://${oc.env:GENFORGE_EXP_ROOT}
+  job:
+    chdir: false
+  output_subdir: null
+  run:
+    dir: .

genforge-0.1.0/src/forge/configs/control/.gitkeep

@@ -0,0 +1 @@
+# Phase 0: empty config group for `control`. Concrete options arrive in later phases.

genforge-0.1.0/src/forge/configs/control/cbf.yaml

@@ -0,0 +1,3 @@
+name: cbf
+params:
+  alpha: 1.0

genforge-0.1.0/src/forge/configs/control/fbsde_control.yaml

@@ -0,0 +1,4 @@
+name: fbsde_control
+params:
+  value_checkpoint: checkpoints/lq/fbsde/values.pt
+  scale: 1.0

genforge-0.1.0/src/forge/configs/control/guidance.yaml

@@ -0,0 +1,4 @@
+name: guidance
+params:
+  scale: 2.0
+  sigma_weight: true

genforge-0.1.0/src/forge/configs/control/projection.yaml

@@ -0,0 +1,2 @@
+name: projection
+params: {}

genforge-0.1.0/src/forge/configs/control/value_guidance.yaml

@@ -0,0 +1,5 @@
+name: value_guidance
+params:
+  value_checkpoint: checkpoints/distributions/value/values.pt
+  scale: 3.0
+  sigma_weight: true

genforge-0.1.0/src/forge/configs/cost/.gitkeep

@@ -0,0 +1 @@
+# Phase 0: empty config group for `cost`. Concrete options arrive in later phases.

genforge-0.1.0/src/forge/configs/cost/barrier.yaml

@@ -0,0 +1,4 @@
+name: barrier
+params:
+  normal: [1.0, 0.0]
+  offset: 0.0

genforge-0.1.0/src/forge/configs/cost/halfspace.yaml

@@ -0,0 +1,4 @@
+name: halfspace
+params:
+  normal: [1.0, 0.0]
+  offset: 0.0

genforge-0.1.0/src/forge/configs/cost/reward.yaml

@@ -0,0 +1,4 @@
+name: reward
+params:
+  target: [2.0, 0.0]
+  weight: 1.0

genforge-0.1.0/src/forge/configs/dataset/.gitkeep

@@ -0,0 +1 @@
+# Phase 0: empty config group for `dataset`. Concrete options arrive in later phases.

genforge-0.1.0/src/forge/configs/environment/.gitkeep

@@ -0,0 +1 @@
+# Phase 0: empty config group for `environment`. Concrete options arrive in later phases.

genforge-0.1.0/src/forge/configs/method/.gitkeep

@@ -0,0 +1 @@
+# Phase 0: empty config group for `method`. Concrete options arrive in later phases.

genforge-0.1.0/src/forge/configs/method/conditional.yaml

@@ -0,0 +1,3 @@
+name: conditional
+params:
+  pin_positions: [0, -1]

genforge-0.1.0/src/forge/configs/method/d3pm.yaml

@@ -0,0 +1,2 @@
+name: d3pm
+params: {}

genforge-0.1.0/src/forge/configs/method/ddpm.yaml

@@ -0,0 +1,2 @@
+name: ddpm
+params: {}

genforge-0.1.0/src/forge/configs/method/ddpm_huber.yaml

@@ -0,0 +1,3 @@
+name: ddpm_huber
+params:
+  delta: 1.0   # residual threshold; |r| > delta is penalized linearly instead of quadratically

genforge-0.1.0/src/forge/configs/method/fbsde.yaml

@@ -0,0 +1,3 @@
+name: fbsde
+params:
+  q: 4.0

genforge-0.1.0/src/forge/configs/method/flow_matching.yaml

@@ -0,0 +1,2 @@
+name: flow_matching
+params: {}

genforge-0.1.0/src/forge/configs/method/mdlm.yaml

@@ -0,0 +1,2 @@
+name: mdlm
+params: {}

genforge-0.1.0/src/forge/configs/method/ot_cfm.yaml

@@ -0,0 +1,2 @@
+name: ot_cfm
+params: {}

genforge-0.1.0/src/forge/configs/method/sedd.yaml

@@ -0,0 +1,2 @@
+name: sedd
+params: {}

genforge-0.1.0/src/forge/configs/method/value_training.yaml

@@ -0,0 +1,4 @@
+name: value_training
+params:
+  target: [2.0, 0.0]
+  weight: 1.0

genforge-0.1.0/src/forge/configs/model/.gitkeep

@@ -0,0 +1 @@
+# Phase 0: empty config group for `model`. Concrete options arrive in later phases.

genforge-0.1.0/src/forge/configs/model/categorical_mlp.yaml

@@ -0,0 +1,5 @@
+name: categorical_mlp
+params:
+  num_classes: 5
+  hidden: 128
+  depth: 3

genforge-0.1.0/src/forge/configs/model/mlp.yaml

@@ -0,0 +1,6 @@
+name: mlp
+params:
+  dim: 2
+  hidden: 128
+  depth: 3
+  output_type: eps

genforge-0.1.0/src/forge/configs/model/temporal_unet.yaml

@@ -0,0 +1,6 @@
+name: temporal_unet
+params:
+  dim: 2
+  horizon: 32
+  base: 32
+  output_type: eps

genforge-0.1.0/src/forge/configs/model/temporal_unet_janner.yaml

@@ -0,0 +1,10 @@
+name: temporal_unet_janner
+params:
+  dim: 32
+  dim_mults: [1, 2, 4, 8]
+  horizon: 32
+  transition_dim: 14
+  cond_dim: 0
+  cond_predict_scale: false
+  output_type: x0
+  obs_dim: 0