BrainPy icon indicating copy to clipboard operation
BrainPy copied to clipboard

Published 20 hours ago • verified publisher icon brainpy

[math] Update `CustomOpByNumba` to support JAX version >= 0.4.24

Open Routhleck opened this issue 1 year ago • 0 comments

Update CustomOpByNumba to support JAX version >= 0.4.24 by implementing mlir register rules

Routhleck avatar May 08 '24 05:05 Routhleck