Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Refactoring of ops.py: code is in general cleaner, and JAXOp can now …
…be used without the decorator @as_jax_op
- Loading branch information