JaxUED: A simple and useable UED library in Jax

Coward, Samuel ; Beukman, Michael ; et al.

2024

Online report

Zugriff:

We present JaxUED, an open-source library providing minimal dependency implementations of modern Unsupervised Environment Design (UED) algorithms in Jax. JaxUED leverages hardware acceleration to obtain on the order of 100x speedups compared to prior, CPU-based implementations. Inspired by CleanRL, we provide fast, clear, understandable, and easily modifiable implementations, with the aim of accelerating research into UED. This paper describes our library and contains baseline results. Code can be found at https://github.com/DramaCow/jaxued.

Comment: 11 pages, 5 figures

Titel:	JaxUED: A simple and useable UED library in Jax
Autor/in / Beteiligte Person:	Coward, Samuel ; Beukman, Michael ; Foerster, Jakob
Link:	View this record from Arxiv (Volltext)
Veröffentlichung:	2024
Medientyp:	report
Schlagwort:	Computer Science - Machine Learning Computer Science - Artificial Intelligence
Sonstiges:	Nachgewiesen in: arXiv Collection: Computer Science Document Type: Working Paper

Klicken Sie ein Format an und speichern Sie dann die Daten oder geben Sie eine Empfänger-Adresse ein und lassen Sie sich per Email zusenden.

BibTeX Citavi, JabRef, u.a.
(Literaturverwaltung)

PDF kein Volltext!
(Merkzettel, Notizen)

RIS Endnote, Citavi u.a.
(Literaturverwaltung)

MODS
(XML zur Weiterverarbeitung)

oder

Wählen Sie das für Sie passende Zitationsformat und kopieren Sie es dann in die Zwischenablage, lassen es sich per Mail zusenden oder speichern es als PDF-Datei.

Gewünschter Zitations-Stil:

oder

Bitte prüfen Sie, ob die Zitation formal korrekt ist, bevor Sie sie in einer Arbeit verwenden. Benutzen Sie gegebenenfalls den "Exportieren"-Dialog, wenn Sie ein Literaturverwaltungsprogramm verwenden und die Zitat-Angaben selbst formatieren wollen.