We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 20819c4 commit 60b3e89Copy full SHA for 60b3e89
brainpy/_src/analysis/lowdim/lowdim_analyzer.py
@@ -436,7 +436,7 @@ def F_fy(self):
436
f = partial(f, **(self.pars_update + self.fixed_vars))
437
f = utils.f_without_jaxarray_return(f)
438
f = utils.remove_return_shape(f)
439
- self.analyzed_results[C.F_fy] = bm.jit(f, device=self.jit_device)
+ self.analyzed_results[C.F_fy] = jax.jit(f, device=self.jit_device)
440
return self.analyzed_results[C.F_fy]
441
442
@property
0 commit comments