-
Notifications
You must be signed in to change notification settings - Fork 35
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
Make shallow circuit demo Catalyst-Compatible #816
Conversation
Hello. You may have forgotten to update the changelog!
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @tzunghanjuang !
Just a note that codecov/project coverage failure usually means (a) the feature branch's new feature is not covered by a new test and/or (b) the feature branch is out of sync with main branch. |
8293cc5
to
6daacb7
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice work on the demo, it seems like it uncovered quite a few issues for us so that's great!
Co-authored-by: David Ittah <dime10@users.noreply.github.com>
Co-authored-by: David Ittah <dime10@users.noreply.github.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🎉
Context:
Convert Shallow Circuit Demo into a Catalyst-compatiable one.
Description of the Change: Mainly replace
jax.jit
withqjit
. However, it introduces errors related tofunctools.partial
(#814 and #815). Right now I circumvent the errors by avoiding usingfunctools.partial
. Will open the other PRs to fix the issues.optax
optimizers, we usepytree
to handle such information.optax
from the AutoGraph conversion.Related GitHub Issues:
[sc-65901]