求助：ModuleNotFoundError: No module named 'jax.experimental.vectorize'问题

jax.experimental.vectorize在JAX 0.4.x版本中已被移除或迁移，这就是你在0.4.10版本中遇到该错误的原因。以下是可行的解决办法：

• 改用官方推荐的jax.vmap：这是JAX目前标准的向量化工具，功能覆盖原vectorize的核心场景。将代码中的from jax.experimental.vectorize import vectorize替换为from jax import vmap，调用方式基本一致，例如原vectorize(func)可直接替换为vmap(func)。
• 降级JAX版本至0.3.x系列：如果需要完全兼容原代码，可安装仍保留该模块的旧版本，执行命令：pip install jax==0.3.25
• 排查其他API兼容问题：他人编写的代码大概率基于旧版JAX开发，除vectorize外可能存在其他API变动，建议对照JAX版本更新日志逐一确认调整。

内容的提问来源于stack exchange，提问作者Cyebukayire