atjerg

其初始版本要求 jax[cuda12, tpu]>=0.7.2 (github.com/openxla/tokamax/commit/38628ea6ddfe6429215a751de36a1fd963d4e0a4), 是否意味着目前无法运行.

目前沐曦提供的 jax 版本只有 0.4.34 , 在需要配置 tokamax 的时候, 二者会发生冲突, 请问有什么解决办法?