Skip to content

hardware

hardware

Cluster information.

Cluster

Bases: BaseModel

log_dir: str property

Log directory path, defaults to {work}/logs if not configured.

HardwareRequest

Bases: BaseModel

Minimal, intent-level hardware request.

Assumptions
  • Storage is uniform across hosts (distributed FS).
  • Interconnects / topology are cluster concerns.

HardwareResult

Bases: BaseModel

Concrete hardware allocation result.

local(root_dir: str, work_dir: str) -> HardwareResult

Detect hardware using JAX device information. Creates one ClusterMachine per host, with host index matching jax.process_index(). Assumes paths are identical across all hosts.