Merge branch 'main' into jonbolin/16vm-aot

google · Aug 15, 2024 · cddc5ad · cddc5ad
2 parents e685da1 + 2f58552
commit cddc5ad
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