From d7c7a030cd11c3f9316e83fc271f25f4e3b90faa Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 4 Dec 2025 01:57:25 -0800
Subject: [PATCH] Add JAX and JAXLIB version metadata to xspace

PiperOrigin-RevId: 840138208
---
 tsl/profiler/protobuf/profiler_options.proto | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tsl/profiler/protobuf/profiler_options.proto b/tsl/profiler/protobuf/profiler_options.proto
index 5ffe1e799..6edfeda06 100644
--- a/tsl/profiler/protobuf/profiler_options.proto
+++ b/tsl/profiler/protobuf/profiler_options.proto
@@ -2,7 +2,7 @@ syntax = "proto3";
 
 package tensorflow;
 
-// Next ID: 15
+// Next ID: 17
 message ProfileOptions {
   // Some default value of option are not proto3 default value. Use this version
   // to determine if we should use default option value instead of proto3
@@ -100,6 +100,11 @@ message ProfileOptions {
   // Identifier of the profiling session. This will be used as the subdirectory
   // under the repository path. If not set, the current timestamp will be used.
   string session_id = 14;
+
+  // JAX version.
+  string jax_version = 15;
+  // JAXlib version.
+  string jaxlib_version = 16;
 }
 
 // Options for remote profiler session manager.