Merge pull request #825 from google:proxy_registration

PiperOrigin-RevId: 663098213
AI-Hypercomputer · Aug 14, 2024 · 2f58552 · 2f58552
2 parents f904ede + d786994
commit 2f58552
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 36 deletions.
diff --git a/MaxText/maxengine_server.py b/MaxText/maxengine_server.py
@@ -19,8 +19,6 @@
 import sys
 import pyconfig
 
-# pylint: disable-next=unused-import
-import register_jax_proxy_backend
 import maxengine_config
 from jetstream.core import server_lib, config_lib
 

diff --git a/MaxText/register_jax_proxy_backend.py b/MaxText/register_jax_proxy_backend.py
diff --git a/MaxText/train.py b/MaxText/train.py
@@ -43,8 +43,6 @@
 import optimizers
 import profiler
 import pyconfig
-# pylint: disable-next=unused-import
-import register_jax_proxy_backend
 from vertex_tensorboard import VertexTensorboardManager
 # Placeholder: internal