search

sail-sg / autofd

Automatic Functional Differentiation in JAX
Apache License 2.0
54 stars 1 forks source link

fix nabla for function with different in/out types #4

Closed mavenlin closed 10 months ago