-
Notifications
You must be signed in to change notification settings - Fork 7
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add binary operation specialization #2
Conversation
Pull Request Test Coverage Report for Build 3203863830
💛 - Coveralls |
@quantshah |
Everything looks good here except that we cannot pass QuTiP objects to and from functions that require JIT compilation. I wrote a simple use-case to show this. We probably need to do something like this: But I am not sure how what's the best way to go here. |
Ok I will add the JIT tests in a separate PR. Everything looks good here. Will merge after the tests pass. |
Add binary operations between
JaxArray
:matmul
,add
,sub
,kron
,multiply
,and binary operation between
JaxArray
and scalar:mul
,pow
.