# Kubeflow Trainer v2.2: JAX & XGBoost Runtimes, Flux for HPC Support, and TrainJob progress and metrics observability

DevFeed: [Kubeflow Trainer v2.2: JAX & XGBoost Runtimes, Flux for HPC Support, and TrainJob progress and metrics observability](<https://devfeed.tech/articles/kubeflow-trainer-v2-2-jax-xgboost-runtimes-flux-for-hpc-support-and-trainjob-progress-and-metrics-observability-17612.md>)

Original publisher: [Read original article](<https://blog.kubeflow.org/kubeflow-trainer-v2.2-release/>)

Author: Kubeflow Trainer Team

Published: 2026-03-20T05:00:00Z

Content type: release

Language: en

Sources: [Kubeflow](<https://devfeed.tech/sources/kubeflow.md>)

Topics: [Kubernetes](<https://devfeed.tech/topics/kubernetes.md>), [distributed-training](<https://devfeed.tech/topics/distributed-training.md>), [observability](<https://devfeed.tech/topics/observability.md>), [flux](<https://devfeed.tech/topics/flux.md>), [Fine-tuning](<https://devfeed.tech/topics/fine-tuning.md>)

Tags: [distributed-training](<https://devfeed.tech/tags/distributed-training.md>), [fine-tuning](<https://devfeed.tech/tags/fine-tuning.md>), [flux](<https://devfeed.tech/tags/flux.md>), [gpu](<https://devfeed.tech/tags/gpu.md>), [hugging-face](<https://devfeed.tech/tags/hugging-face.md>), [kubernetes](<https://devfeed.tech/tags/kubernetes.md>), [observability](<https://devfeed.tech/tags/observability.md>), [release](<https://devfeed.tech/tags/release.md>), [trainer](<https://devfeed.tech/tags/trainer.md>)

## AI overview

Kubeflow Trainer v2.2 adds native JAX and XGBoost training runtimes, Flux runtime support for HPC workloads, enhanced training-job observability, timeout policies, and more flexible runtime configuration through new APIs. It also unifies supported training workloads under the TrainJob abstraction and enables distributed JAX workloads on Kubernetes.

## Source excerpt

Just a little over one week ahead of KubeCon + CloudNativeCon EU 2026, the Kubeflow team is excited to ship Trainer v2.2. The v2.2 release reinforces our commitment to expanding the Kubeflow Trainer ecosystem - meeting developers where they are by adding native support for JAX, XGBoost, and Flux, while also delivering deeper observability into training jobs.