jax

Documentation

Index

• Constants
• type JaxGrouper
  • func NewJaxGrouper(kubeflowGrouper *kubeflow.KubeflowDistributedGrouper) *JaxGrouper
  • func (jg *JaxGrouper) GetPodGroupMetadata(topOwner *unstructured.Unstructured, pod *v1.Pod, ...) (*podgroup.Metadata, error)
  • func (jg *JaxGrouper) Name() string

Constants

const (
	ReplicaSpecName = "jaxReplicaSpecs"
	WorkerName      = "Worker"
)

Types

type JaxGrouper

type JaxGrouper struct {
	*kubeflow.KubeflowDistributedGrouper
}

func NewJaxGrouper

func NewJaxGrouper(kubeflowGrouper *kubeflow.KubeflowDistributedGrouper) *JaxGrouper

func (*JaxGrouper) GetPodGroupMetadata

func (jg *JaxGrouper) GetPodGroupMetadata(
	topOwner *unstructured.Unstructured, pod *v1.Pod, _ ...*metav1.PartialObjectMetadata,
) (*podgroup.Metadata, error)

func (*JaxGrouper) Name

func (jg *JaxGrouper) Name() string