Duck - a package for the duck (operator learning) renormalization group

This package is a Python implementation of the duck (operator learning) renormalization group in JAX.

[!NOTE] There was an older implementation of the duck RG in teal, which was used for the first 2 versions of the duck RG theory paper.

[!IMPORTANT] This package is still under development in alpha stage, and the API may change in the future.

Installation

This package is available on PyPI, thus can be installed via pip:

pip install rogerluo-duck

However, we highly recommend using uv to install the package, run the following in your Python project.

uv add rogerluo-duck

Features

• a simple symbolic system for defining operators
• a set of differentiable local solvers defined on top of the above symbolic system
• implementation of the duck RG loss function
• a set of utilities for training and evaluating the machine learning model in the duck RG

Documentation

The documentation is available at https://rogerluo.dev/duck/.

License

This package is licensed under the Apache License 2.0.