Cover Image

JaxUED: A simple and useable UED library in Jax

Fuente: arXiv

Saved in:

Bibliographic Details
Main Authors:	Coward, Samuel, Beukman, Michael, Foerster, Jakob
Format:	Preprint
Published:	2024
Subjects:	Machine Learning Artificial Intelligence
Online Access:	Acceder al recurso
Tags:	Add Tag No Tags, Be the first to tag this record!

Similar Items

JaxLife: An Open-Ended Agentic Simulator
by: Lu, Chris, et al.
Published: (2024)

JaxMARL: Multi-Agent RL Environments and Algorithms in JAX
by: Rutherford, Alexander, et al.
Published: (2023)

Refining Minimax Regret for Unsupervised Environment Design
by: Beukman, Michael, et al.
Published: (2024)

JaxWildfire: A GPU-Accelerated Wildfire Simulator for Reinforcement Learning
by: Çakır, Ufuk, et al.
Published: (2025)

Kinetix: Investigating the Training of General Agents through Open-Ended Physics-Based Control Tasks
by: Matthews, Michael, et al.
Published: (2024)

JaxARC: A High-Performance JAX-based Environment for Abstraction and Reasoning Research
by: Aadam, et al.
Published: (2026)

SocialJax: An Evaluation Suite for Multi-agent Reinforcement Learning in Sequential Social Dilemmas
by: Guo, Zihao, et al.
Published: (2025)

EconoJax: A Fast & Scalable Economic Simulation in Jax
by: Ponse, Koen, et al.
Published: (2024)

No Regrets: Investigating and Improving Regret Approximations for Curriculum Discovery
by: Rutherford, Alexander, et al.
Published: (2024)

DéjàQ: Open-Ended Evolution of Diverse, Learnable and Verifiable Problems
by: Röpke, Willem, et al.
Published: (2026)

Goal-Conditioned Agents that Learn Everything All at Once
by: Matthews, Michael, et al.
Published: (2026)

Craftax: A Lightning-Fast Benchmark for Open-Ended Reinforcement Learning
by: Matthews, Michael, et al.
Published: (2024)

JaxSGMC: Modular stochastic gradient MCMC in JAX
by: Thaler, Stephan, et al.
Published: (2025)

Chronax: A Jax Library for Univariate Statistical Forecasting and Conformal Inference
by: Carey, Xan, et al.
Published: (2026)

JaxDecompiler: Redefining Gradient-Informed Software Design
by: Pochelu, Pierrick
Published: (2024)

NASimJax: GPU-Accelerated Policy Learning Framework for Penetration Testing
by: Simon, Raphael, et al.
Published: (2026)

JaxRobotarium: Training and Deploying Multi-Robot Policies in 10 Minutes
by: Jain, Shalin Anand, et al.
Published: (2025)

Learning Multi-Agent Communication with Contrastive Learning
by: Lo, Yat Long, et al.
Published: (2023)

Mixture of Experts in a Mixture of RL settings
by: Willi, Timon, et al.
Published: (2024)

Mirror Learning: A Unifying Framework of Policy Optimisation
by: Kuba, Jakub Grudzien, et al.
Published: (2022)

Magnetic Compressors for MeV-UED
by: Xu, Tianzhe, et al.
Published: (2024)

The Edge-of-Reach Problem in Offline Model-Based Reinforcement Learning
by: Sims, Anya, et al.
Published: (2024)

Symmetry-Breaking Augmentations for Ad Hoc Teamwork
by: Hammond, Ravi, et al.
Published: (2024)

Abstraction for Offline Goal-Conditioned Reinforcement Learning
by: Wibault, Clarisse, et al.
Published: (2026)

A Model-Based Solution to the Offline Multi-Agent Reinforcement Learning Coordination Problem
by: Barde, Paul, et al.
Published: (2023)

Behaviour Distillation
by: Lupu, Andrei, et al.
Published: (2024)

DITTO: Offline Imitation Learning with World Models
by: DeMoss, Branton, et al.
Published: (2023)

SOReL and TOReL: Two Methods for Fully Offline Reinforcement Learning
by: Fellows, Mattie, et al.
Published: (2025)

Hierarchical Behaviour Spaces
by: Matthews, Michael Tryfan, et al.
Published: (2026)

Beyond the Boundaries of Proximal Policy Optimization
by: Tan, Charlie B., et al.
Published: (2024)

Rethinking Out-of-Distribution Detection for Reinforcement Learning: Advancing Methods for Evaluation and Detection
by: Nasvytis, Linas, et al.
Published: (2024)

Recurrent Reinforcement Learning with Memoroids
by: Morad, Steven, et al.
Published: (2024)

Meta-Learning Objectives for Preference Optimization
by: Alfano, Carlo, et al.
Published: (2024)

How Should We Meta-Learn Reinforcement Learning Algorithms?
by: Goldie, Alexander David, et al.
Published: (2025)

High entropy leads to symmetry equivariant policies in Dec-POMDPs
by: Forkel, Johannes, et al.
Published: (2025)

A Clean Slate for Offline Reinforcement Learning
by: Jackson, Matthew Thomas, et al.
Published: (2025)

Can Learned Optimization Make Reinforcement Learning Less Difficult?
by: Goldie, Alexander David, et al.
Published: (2024)

Assessing the Zero-Shot Capabilities of LLMs for Action Evaluation in RL
by: Pignatelli, Eduardo, et al.
Published: (2024)

Policy-Guided Diffusion
by: Jackson, Matthew Thomas, et al.
Published: (2024)

$\texttt{lrnnx}$: A library for Linear RNNs
by: Bania, Karan, et al.
Published: (2026)