removed jax distributed initialization

ajkv-google · ajkv-google · commit e97f4fb49098 · 2026-01-22T22:19:27.000Z
diff --git a/recml/examples/dlrm_experiment_test.py b/recml/examples/dlrm_experiment_test.py
@@ -15,9 +15,6 @@
 
 import jax
 from absl import logging
-
-jax.distributed.initialize()
-
 import sys
 import os
 # Add the RecML folder to the system path
