§ W’jãó>—ddlmZddlmZddlmZmZmZddlm Z dS)é)Ú chunk_vmap)Úreplace_all_batch_norm_modules_)ÚhessianÚjacfwdÚjvp)Ú functionalizeN) Útorch._functorch.apisrÚ'torch._functorch.batch_norm_replacementrÚ!torch._functorch.eager_transformsrrrÚ torch.funcr©óúd/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/functorch/experimental/__init__.pyúrs_ðà,Ð,Ð,Ð,Ð,Ð,ØSÐSÐSÐSÐSÐSØBÐBÐBÐBÐBÐBÐBÐBÐBÐBØ$Ð$Ð$Ð$Ð$Ð$Ð$Ð$r