Repository Map¶
AM-Bench separates simulation, imitation learning, command-line workflows, internal engineering notes, and third-party integrations.
am_isaac/
source/am_isaac/ simulation extension
source/am_isaac_il/ learned-policy and data adapters
scripts/environments/ registration, smoke tests, teleoperation
scripts/data/ recording, validation, format conversion
docs/ public documentation
ext/ optional external dependencies
Simulation package¶
| Path | Responsibility |
|---|---|
tasks/ |
Scene setup, observations, rewards, success, and reset logic |
tasks/base/ |
Shared BaseEnv, configuration, registrations, and robot profiles |
scenes/ |
Reusable task-agnostic scene components |
robots/ |
Morphology, assets, rotor layouts, and RobotIO semantics |
controllers/ |
Action pipelines, IK integration, wrench controllers, and allocation |
disturbance/ |
Aerodynamic and disturbance models |
policies/scripted/ |
Task-aware scripted policies |
recording/ |
Canonical LeRobot and video writers |
evaluation/ |
Controller tracking and limit-use diagnostics |
Imitation-learning package¶
| Path | Responsibility |
|---|---|
data/ |
Logical resampling and action-semantic validation |
pipeline/ |
Dataset and action transformations shared by learned policies |
eval/ |
Common evaluator CLI, batching, outputs, and tracking lifecycle |
policies/act/ |
ACT datasets, processor, model, training, and evaluation |
policies/dp/ |
Diffusion Policy integration and UMI dependency |
policies/pi/ |
OpenPI client-side evaluation adapter |
The ext/pyroki and ext/acados trees are external integrations. Changes to them should be isolated from ordinary task or documentation work.
Development-only datasets, checkpoints, videos, experiment wrappers, and generated build outputs are not part of the public package boundary.