diff --git a/tsl/profiler/protobuf/profiler_options.proto b/tsl/profiler/protobuf/profiler_options.proto
index 04c46af2d..587420d6b 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: 16
+// Next ID: 18
 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
@@ -103,6 +103,11 @@ message ProfileOptions {
 
   // If set, this hostname will be used to name the profile file.
   string override_hostname = 15;
+
+  // JAX version.
+  string jax_version = 16;
+  // JAXlib version.
+  string jaxlib_version = 17;
 }
 
 // Options for remote profiler session manager.