Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/Roger-luo/teal
Software framework for experimenting with Operator Learning Renormalization Group (OLRG)
https://github.com/Roger-luo/teal
Last synced: 11 days ago
JSON representation
Software framework for experimenting with Operator Learning Renormalization Group (OLRG)
- Host: GitHub
- URL: https://github.com/Roger-luo/teal
- Owner: Roger-luo
- License: other
- Created: 2024-02-02T18:26:23.000Z (9 months ago)
- Default Branch: main
- Last Pushed: 2024-03-06T05:02:48.000Z (8 months ago)
- Last Synced: 2024-06-27T21:20:07.557Z (5 months ago)
- Language: Python
- Size: 68.4 KB
- Stars: 2
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Teal
A Python Package for Operator Learning Renromalization Group.
**Warning**: This package is currently at its early stages of development. It is not yet ready for serious use. All the APIs are subject to change. No unit tests are available yet. However the implementation has been tested in private projects and is expected to work.
This package contains the implementation for paper [Operator Learning Renormalization Group, arXiv:2403.03199](https://arxiv.org/abs/2403.03199) as [paper-v1 release](https://github.com/Roger-luo/teal/releases/tag/paper-v1).
## Installation
This package is currently at its early stages of development. Thus it is not yet available on PyPI. We use [rye](https://rye-up.com/) for package management. To install it from source, clone the repository and run:
```sh
rye sync
```To setup CUDA environment, please refer to [jax documents](https://jax.readthedocs.io/en/latest/).
## Examples
Running HEM example
```sh
python examples/hem.py --wandb=False\
--n-iterations=5000\
--ham=TFIM\
--n-start=2\
--n-final=6\
--enlarge-by=1\
--final-time=0.1\
--order=2\
--n-batch=1\
--depth=4\
--width=4\
--order-factor=one\
--n-samples=20
```Running OMM example
**Warning**: This may require a GPU to run.
```sh
python examples/omm.py --wandb=False --n-iterations=5000 --ham=TFIM --n-start=4 --n-final=10 --enlarge-by=1 --final-time=0.1 --order=2 --n-batch=5 --depth=8 --order-factor=one --n-samples=20
```## License
Apache License 2.0