feat(runtimes): KEP-2442-jax-runtime

[mahdikhashan]

What this PR does / why we need it:

Which issue(s) this PR fixes (optional, in Fixes #<issue number>, #<issue number>, ... format, will close the issue(s) when PR gets merged):
Fixes #

Checklist:

• Docs included if any changes are user facing

[google-oss-prow]

[APPROVALNOTIFIER] This PR is NOT APPROVED

This pull-request has been approved by:
Once this PR has been reviewed and has the lgtm label, please assign johnugeorge for approval. For more information see the Kubernetes Code Review Process.

The full list of commands accepted by this bot can be found here.

Needs approval from an approver in each of these files:

• OWNERS

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

[andreyvelich]

Thanks for this awesome work @mahdikhashan!
Please let us know if this PR is ready for review.

[andreyvelich]

/milestone v2.1

[andreyvelich]

@mahdikhashan @Electronic-Waste Is this PR ready for review ?
We are planning to release v2.1.0-rc.0, and we would like to know whether we should target this feature in v2.1?

[mahdikhashan]

@mahdikhashan @Electronic-Waste Is this PR ready for review ? We are planning to release v2.1.0-rc.0, and we would like to know whether we should target this feature in v2.1?

what is the estimated deadline for the release?

[andreyvelich]

We are planning to release v2.1.0-rc.0 this week, the official release should be ready in ~ 2 weeks.

[andreyvelich]

@mahdikhashan @Electronic-Waste Given that we released the Trainer v2.1 recently, do you want to finalize this work to support JAX runtime in TrainJob ?

[mahdikhashan]

@mahdikhashan @Electronic-Waste Given that we released the Trainer v2.1 recently, do you want to finalize this work to support JAX runtime in TrainJob ?

sorry for this, i got busy with my master's practical work. I'll try to work on it in december break, hope its fine.