Skip to content
This repository has been archived by the owner on May 11, 2023. It is now read-only.

feat: Will "Custom Linear Operator" also work for "black box" code? #28

Open
mreineck opened this issue Mar 29, 2023 · 0 comments
Open
Labels
enhancement New feature or request

Comments

@mreineck
Copy link

I'm working on a collection of optimized linear operators, including FFTs, non-uniform FFTs, spherical harmonic transforms etc. (see https://github.com/mreineck/ducc if you are interested in more details). These are implemented in C++, but callable through a simple Python interface. For every operator I'm providing forward application as well as the adjoint.
It would be great to make Jax aware of these operations, but at least at the moment there doesn't seem to be an easy way to do this. Will your "Custom Linear Operator" enhancement be able to support this kind of code, or will it be limited to Python code that is fully inspectable by Jax?

@mreineck mreineck added the enhancement New feature or request label Mar 29, 2023
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant