Ë ñukiãó,—ddlmZmZddlmZmZmZmZy)é)Úcustom_partitioningÚcustom_partitioning_p)ÚBATCHINGÚCompoundFactorÚ ArrayMappingÚSdyShardingRuleN)Újax._src.custom_partitioningrrÚ*jax._src.custom_partitioning_sharding_rulerrrr©óú_/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/experimental/custom_partitioning.pyúrsð÷$÷ ôr