mirror of
https://github.com/hanzo-ml/trainer.git
synced 2026-09-24 00:31:34 +00:00
696 B
696 B
KubeflowOrgV1JAXJobSpec
JAXJobSpec is a desired state description of the JAXJob.
Properties
| Name | Type | Description | Notes |
|---|---|---|---|
| jax_replica_specs | dict(str, KubeflowOrgV1ReplicaSpec) | A map of JAXReplicaType (type) to ReplicaSpec (value). Specifies the JAX cluster configuration. For example, { "Worker": JAXReplicaSpec, } | |
| run_policy | KubeflowOrgV1RunPolicy |