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.