Skip to main content

Transformer Metacontroller

Project description

metacontroller (wip)

Implementation of the MetaController proposed in Emergent temporal abstractions in autoregressive models enable hierarchical reinforcement learning

Install

$ pip install metacontroller-pytorch

Appreciation

  • Pranoy for submitting a pull request for fixing the previous latent action not being included in the inputs to the switching unit

  • Diego Calanzone for proposing testing on BabyAI gridworld task, and submitting the pull request for behavior cloning and discovery phase training for it!

Citations

@misc{kobayashi2025emergenttemporalabstractionsautoregressive,
    title   = {Emergent temporal abstractions in autoregressive models enable hierarchical reinforcement learning}, 
    author  = {Seijin Kobayashi and Yanick Schimpf and Maximilian Schlegel and Angelika Steger and Maciej Wolczyk and Johannes von Oswald and Nino Scherrer and Kaitlin Maile and Guillaume Lajoie and Blake A. Richards and Rif A. Saurous and James Manyika and Blaise Agüera y Arcas and Alexander Meulemans and João Sacramento},
    year    = {2025},
    eprint  = {2512.20605},
    archivePrefix = {arXiv},
    primaryClass = {cs.LG},
    url     = {https://arxiv.org/abs/2512.20605}, 
}
@article{Wagenmaker2025SteeringYD,
    title   = {Steering Your Diffusion Policy with Latent Space Reinforcement Learning},
    author  = {Andrew Wagenmaker and Mitsuhiko Nakamoto and Yunchu Zhang and Seohong Park and Waleed Yagoub and Anusha Nagabandi and Abhishek Gupta and Sergey Levine},
    journal = {ArXiv},
    year    = {2025},
    volume  = {abs/2506.15799},
    url     = {https://api.semanticscholar.org/CorpusID:279464702}
}
@misc{fleuret2025freetransformer,
    title     = {The Free Transformer}, 
    author    = {François Fleuret},
    year      = {2025},
    eprint    = {2510.17558},
    archivePrefix = {arXiv},
    primaryClass = {cs.LG},
    url       = {https://arxiv.org/abs/2510.17558}, 
}

Life can only be understood backwards; but it must be lived forwards - Søren Kierkegaard

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

metacontroller_pytorch-0.0.31.tar.gz (335.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

metacontroller_pytorch-0.0.31-py3-none-any.whl (12.8 kB view details)

Uploaded Python 3

File details

Details for the file metacontroller_pytorch-0.0.31.tar.gz.

File metadata

  • Download URL: metacontroller_pytorch-0.0.31.tar.gz
  • Upload date:
  • Size: 335.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.9.25

File hashes

Hashes for metacontroller_pytorch-0.0.31.tar.gz
Algorithm Hash digest
SHA256 7bc721b283c5fafb2e1acfb678b349db28640400a5e8a989636db68df2038d38
MD5 b5b1ac1ba4a2e4fefe9d58875c51f19f
BLAKE2b-256 ac9b9782a21d41500ee73d126f65aa6bf2d69361c20e48f823a810f2655a6f49

See more details on using hashes here.

File details

Details for the file metacontroller_pytorch-0.0.31-py3-none-any.whl.

File metadata

File hashes

Hashes for metacontroller_pytorch-0.0.31-py3-none-any.whl
Algorithm Hash digest
SHA256 0b6fd9549b0c19e40663b2c496e7fb6874ffaa1a772e8e4c917071e2495ce120
MD5 46ff4bed5e3bb85e43fa4e882bb75c50
BLAKE2b-256 3428698c4859302be370e4f0bb25b63c69f06877e4a9a4770e77ba0e329db17f

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page