jax
bcad02ff - Remove 64-bit mode for GPU performance

Commit

6 years ago

Remove 64-bit mode for GPU performance

References

#1255 - Pass check_vjp, add VJP grad and Jac, use fun(y0, t, *args) syntax

Author

Daniel Weaver

Parents

Loading