tony/tensorneat-mend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
41 Commits 2 Branches 0 Tags
6006f92f3f07adbcc7a4ad93284532fff98b0ea0
Commit Graph

2 Commits

Author SHA1 Message Date
wls2002
6006f92f3f finish jit-able speciate function 
next time i'll create a new branch
 2023-05-12 19:26:02 +08:00
wls2002
9b56f4ff73 做了一些时间测试： 
1. jit中的vmap一个函数不会触发重新编译
2. jax.lax.while_lop单独执行确实可以提前中断，但在vmap中的性能需要考虑。
 2023-05-12 16:42:57 +08:00