XPK Start: Sat Apr 25 21:59:25 UTC 2026 PyTorch was not found. Models won't be available and only tokenizers, configuration and file/data utilities can be used. `rope_parameters`'s factor field must be a float >= 1, got 40 `rope_parameters`'s beta_fast field must be a float, got 32 `rope_parameters`'s beta_slow field must be a float, got 1 DeepseekV32Config got `key=rope_scaling` in kwargs but hasn't set it as attribute. For RoPE standardization you need to set `self.rope_parameters` in model's config. 2026-04-25 21:59:50.150185: E external/local_xla/xla/stream_executor/cuda/cuda_platform.cc:51] failed call to cuInit: INTERNAL: CUDA error: Failed call to cuInit: UNKNOWN ERROR (303) I0425 21:59:50.365674 135553364285248 max_utils.py:273] Attempting to initialize the jax distributed system... I0425 21:59:59.407289 135553364285248 distributed.py:149] Starting JAX distributed service on [::]:8482 I0425 21:59:59.409613 135553364285248 distributed.py:172] Connecting to JAX distributed service on mt-14-async-ckpt-false--rq0q9-slice-job-0-0.mt-14-async-ckpt-false--rq0q9:8482 I0425 22:00:00.322546 135553364285248 max_utils.py:284] Jax distributed system initialized! I0425 22:00:06.659606 135553364285248 max_utils.py:800] System Information: Jax Version: 0.9.2 I0425 22:00:06.659715 135553364285248 max_utils.py:801] System Information: Jaxlib Version: 0.9.2 I0425 22:00:06.659778 135553364285248 max_utils.py:802] System Information: Jax Backend: PJRT C API TFRT TPU v6 lite Built on Mar 4 2026 11:32:08 (1772652728) cl/878335365 I0425 22:00:06.659816 135553364285248 train_utils.py:361] WARNING: Sequence packing is essentially ignored for synthetic data. Please use a real dataset to use sequence packing. Traceback (most recent call last): File "<frozen runpy>", line 198, in _run_module_as_main File "<frozen runpy>", line 88, in _run_code File "/deps/src/maxtext/trainers/pre_train/train.py", line 744, in <module> app.run(main) File "/usr/local/lib/python3.12/site-packages/absl/app.py", line 367, in run _run_main(main, args) File "/usr/local/lib/python3.12/site-packages/absl/app.py", line 312, in _run_main sys.exit(main(argv)) ^^^^^^^^^^ File "/deps/src/maxtext/trainers/pre_train/train.py", line 740, in main train_func() File "/deps/src/maxtext/trainers/pre_train/train.py", line 730, in train_func run(config, recorder, diagnostic_config) File "/deps/src/maxtext/trainers/pre_train/train.py", line 709, in run train_loop(config, recorder) File "/deps/src/maxtext/trainers/pre_train/train.py", line 536, in train_loop ) = train_utils.setup_train_loop(config, recorder) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/deps/src/maxtext/utils/train_utils.py", line 218, in setup_train_loop raise NotImplementedError("Pure NNX support has not been implemented yet.") NotImplementedError: Pure NNX support has not been implemented yet. XPK End: Sat Apr 25 22:00:15 UTC 2026 EXIT_CODE=1