qml icon indicating copy to clipboard operation
qml copied to clipboard

Add a how-to for catalyst-compiling "Symmetry-invariant quantum machine learning force fields"

Open paul0403 opened this issue 1 year ago • 16 comments

The how-to contains the full code listing, and some primitive tutorial words.

Before submitting

Please complete the following checklist when submitting a PR:

  • [x] Ensure that your tutorial executes correctly, and conforms to the guidelines specified in the README.

  • [x] Remember to do a grammar check of the content you include.

  • [x] All tutorials conform to PEP8 standards. To auto format files, simply pip install black, and then run black -l 100 path/to/file.py.

When all the above are checked, delete everything above the dashed line and fill in the pull request template.


Title: How to Quantum Just-In-Time Compile "Symmetry-invariant quantum machine learning force fields" with Catalyst

Summary: As part of Catalyst's work on identifying 10 demos to compile with catalyst, we convert "Symmetry-invariant quantum machine learning force fields", a machine learning workflow that calls a training step function repeatedly and thus can have significant performance boosts with catalyst compilation.

Note: the original demo is from an external user, so we should NOT merge this until we have reached out to them. Update: we have obtained their permission.

Relevant references:

Possible Drawbacks: Because lightning does not support differentiation through QubitUnitary, we have to resort to finite difference method for gradients, which brings a big performance degradation.

Related GitHub Issues: [sc-72938]


If you are writing a demonstration, please answer these questions to facilitate the marketing process.

  • GOALS — Why are we working on this now?

    Promote Catalyst by demonstrating the performance improvements it offers by QJIT compiling a relatively complex end-to-end quantum workflow.

  • AUDIENCE — Who is this for?

    Chemistry and quantum machine learning researchers.

  • KEYWORDS — What words should be included in the marketing post?

    • Quantum machine learning
    • Catalyst
    • QJIT
  • Which of the following types of documentation is most similar to your file? (more details here)

  • [ ] Tutorial
  • [ ] Demo
  • [x] How-to

paul0403 avatar Sep 19 '24 15:09 paul0403