Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/kubernetes-sigs/jobset

JobSet: An API for managing a group of Jobs as a unit
https://github.com/kubernetes-sigs/jobset

batch k8s k8s-sig-apps kubernetes

Last synced: about 1 month ago
JSON representation

JobSet: An API for managing a group of Jobs as a unit

Host: GitHub
URL: https://github.com/kubernetes-sigs/jobset
Owner: kubernetes-sigs
License: apache-2.0
Created: 2023-03-31T04:38:12.000Z (over 1 year ago)
Default Branch: main
Last Pushed: 2024-04-13T23:04:32.000Z (8 months ago)
Last Synced: 2024-04-14T08:03:45.448Z (8 months ago)
Topics: batch, k8s, k8s-sig-apps, kubernetes
Language: Go
Homepage: https://jobset.sigs.k8s.io/
Size: 19.2 MB
Stars: 93
Watchers: 7
Forks: 30
Open Issues: 41
Metadata Files:
- Readme: README.md
- Changelog: CHANGELOG/README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
- Code of conduct: code-of-conduct.md
- Security: SECURITY.md

Awesome Lists containing this project

awesome-repositories - kubernetes-sigs/jobset - JobSet: a k8s native API for distributed ML training and HPC workloads (Go)

README

# JobSet

[![GoReport Widget]][GoReport Status]
[![Latest Release](https://img.shields.io/github/v/release/kubernetes-sigs/jobset?include_prereleases)](https://github.com/kubernetes-sigs/jobset/releases/latest)

[GoReport Widget]: https://goreportcard.com/badge/github.com/kubernetes-sigs/jobset
[GoReport Status]: https://goreportcard.com/report/github.com/kubernetes-sigs/jobset

JobSet is a Kubernetes-native API for managing a group of [k8s Jobs](https://kubernetes.io/docs/concepts/workloads/controllers/job/) as a unit. It aims to offer a unified API for deploying HPC (e.g., MPI) and AI/ML training workloads (PyTorch, Jax, Tensorflow etc.) on Kubernetes.

Take a look at the [concepts](https://jobset.sigs.k8s.io/docs/concepts/) page for a brief description of how to use JobSet.

## Conceptual Diagram
jobset diagram

## Features overview

- **Support for multi-template jobs**: JobSet models a distributed training workload as a group of K8s Jobs. This allows a user to easily specify different pod templates for different distinct groups of pods (e.g. a leader, workers, parameter servers, etc.), something which cannot be done by a single Job.

- **Automatic headless service configuration and lifecycle management**: ML and HPC frameworks require a stable network endpoint for each worker in the distributed workload, and since pod IPs are dynamically assigned and can change between restarts, stable pod hostnames are required for distributed training on k8s, By default, JobSet uses [IndexedJobs](https://kubernetes.io/blog/2021/04/19/introducing-indexed-jobs/) to establish stable pod hostnames, and does automatic configuration and lifecycle management of the headless service to trigger DNS record creations and establish network connectivity via pod hostnames. These networking configurations are defaulted automatically to enable stable network endpoints and pod-to-pod communication via hostnames; however, they can be customized in the JobSet spec: see this [example](examples/simple/jobset-with-network.yaml) of using a custom subdomain your JobSet's network configuration.

- **Configurable success policies**: JobSet has [configurable success policies](https://github.com/kubernetes-sigs/jobset/blob/v0.5.0/examples/simple/success-policy.yaml) which target specific ReplicatedJobs, with operators to target `Any` or `All` of their child jobs. For example, you can configure the JobSet to be marked complete if and only if all pods that are part of the “worker” ReplicatedJob are completed. This enables users to use their compute resources more efficiently, allowing a workload to be declared successful and release the resources for the next workload more quickly.

- **Configurable failure policies**: JobSet has [configurable failure policies](https://github.com/kubernetes-sigs/jobset/blob/v0.5.0/examples/simple/max-restarts.yaml) which allow the user to specify a maximum number of times the JobSet should be restarted in the event of a failure. If any job is marked failed, the entire JobSet will be recreated, allowing the workload to resume from the last checkpoint. When no failure policy is specified, if any job fails, the JobSet simply fails.

- **Exclusive Placement Per Topology Domain**: JobSet includes an [annotation](https://github.com/kubernetes-sigs/jobset/blob/1ae6c0c039c21d29083de38ae70d13c2c8ec613f/examples/simple/exclusive-placement.yaml#L6) which can be set by the user, specifying that there should be a 1:1 mapping between child job and a particular topology domain, such as a datacenter rack or zone. This means that all the pods belonging to a child job will be colocated in the same topology domain, while pods from other jobs will not be allowed to run within this domain. This gives the child job exclusive access to computer resources in this domain. You can run this [example](https://github.com/kubernetes-sigs/jobset/blob/v0.5.0/examples/simple/exclusive-placement.yaml) yourself to see how exclusive placement works.

- **Fast failure recovery**: JobSet recovers from failures by recreating all the child Jobs. When scheduling constraints such as exclusive Job placement are used, fast failure recovery at scale can become challenging. As of JobSet v0.3.0, JobSet uses a designed such that it minimizes impact on scheduling throughput. We have benchmarked scheduling throughput during failure recovery at 290 pods/second at a 15k node scale.

- **Startup Sequencing**: As of JobSet v0.5.0 users can configure a [startup order](https://github.com/kubernetes-sigs/jobset/blob/v0.5.0/examples/startup-policy/startup-driver-ready.yaml) for the ReplicatedJobs in a JobSet. This enables support for patterns like the “leader-worker” paradigm, where the leader must be running before the workers should start up and connect to it.

- **Integration with Kueue**: Use JobSet v0.2.3+ and [Kueue](https://kueue.sigs.k8s.io/) v0.5.0+ to oversubscribe your cluster with JobSet workloads, placing them in queue which supports multi-tenancy, resource sharing and more. See [Kueue documentation](https://kueue.sigs.k8s.io/) for more details on the benefits of managing JobSet workloads via Kueue.

## Production Readiness status

- ✔️ API version: v1alpha2, respecting [Kubernetes Deprecation Policy](https://kubernetes.io/docs/reference/using-api/deprecation-policy/)
- ✔️ Maintains support for latest 3 Kubernetes minor versions.
- ✔️ Up-to-date [documentation](https://jobset.sigs.k8s.io/docs).
- ✔️ Test Coverage:
- ✔️ Unit Test [testgrid](https://testgrid.k8s.io/sig-apps#pull-jobset-test-unit-main).
- ✔️ Integration Test [testgrid](https://testgrid.k8s.io/sig-apps#pull-jobset-test-integration-main)
- ✔️ E2E Tests for Kubernetes
[1.27](https://testgrid.k8s.io/sig-apps#pull-jobset-test-e2e-main-1-27),
[1.28](https://testgrid.k8s.io/sig-apps#pull-jobset-test-e2e-main-1-28),
[1.29](https://testgrid.k8s.io/sig-apps#pull-jobset-test-e2e-main-1-29),
[1.30](https://testgrid.k8s.io/sig-apps#pull-jobset-test-e2e-main-1-30)
on Kind.
- ✔️ Monitoring via [metrics](https://jobset.sigs.k8s.io/docs/reference/metrics).
- ✔️ Security: RBAC based accessibility.
- ✔️ Stable release cycle(2-3 months) for new features, bugfixes, cleanups.

## Installation

**Requires a Kubernetes cluster running one of the last 3 Kubernetes minor versions.**

To install the latest release of JobSet in your cluster, run the following command:

```shell
kubectl apply --server-side -f https://github.com/kubernetes-sigs/jobset/releases/download/v0.5.2/manifests.yaml
```

The controller runs in the `jobset-system` namespace.

Read the [installation guide](https://jobset.sigs.k8s.io/docs/installation/) to learn more.

## Roadmap

Below is the JobSet roadmap for 2024, in expected order of release:

- [X] Support for MultiKueue (https://github.com/kubernetes-sigs/jobset/issues/440)
- [X] JobSet TTL to clean up completed workloads (https://github.com/kubernetes-sigs/jobset/issues/279)
- [ ] Configurable Failure Policy (https://github.com/kubernetes-sigs/jobset/issues/262)
- [ ] Graduate API to v1 (https://github.com/kubernetes-sigs/jobset/issues/380)
- [ ] In-place restarts (https://github.com/kubernetes-sigs/jobset/issues/467)
- [ ] Placement Policy API (https://github.com/kubernetes-sigs/jobset/issues/75)
- [ ] Support for JobSet extensions (https://github.com/kubernetes-sigs/jobset/issues/146)
- [ ] PyTorch Extension
- [ ] Jax Extension
- [ ] MPI Extension

## Troubleshooting common issues

See the [troubleshooting](https://jobset.sigs.k8s.io/docs/troubleshooting/) guide for help resolving common issues.

## Community, discussion, contribution, and support

Learn how to engage with the Kubernetes community on the [community page](http://kubernetes.io/community/).

You can reach the maintainers of this project at:

- [Slack](https://kubernetes.slack.com/messages/wg-batch)
- [Mailing List](https://groups.google.com/a/kubernetes.io/g/wg-batch)

### Code of conduct

Participation in the Kubernetes community is governed by the [Kubernetes Code of Conduct](code-of-conduct.md).