[Feature] Support aMUSEd

[okotaku]

Motivation

aMUSEd: An Open MUSE Reproduction

Results (Optional)

amused_512_pokemon_blip

Checklist

Before PR:

• Pre-commit or other linting tools are used to fix the potential lint issues.
• Bug fixes are fully covered by unit tests, the case that causes the bug should be added in the unit tests.
• The modification is covered by complete unit tests. If not, please add more unit test to ensure the correctness.
• The documentation has been modified accordingly, like docstring or example tutorials.

---

📚 Documentation preview 📚: https://DiffEngine--125.org.readthedocs.build/en/125/