PyPI page
Home page
License:
MIT License
Copyright (c) 2025 Quentin Delfosse, Jannis Blüml
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software an...
Summary:
GPU-accelerated, object-centric Atari environments for reinforcement learning with JAX.
Latest version:
0.2.0
Required dependencies:
absl-py
|
ale-py
|
chex
|
flax
|
gymnasium
|
gymnax
|
jax
|
ml-dtypes
|
numpy
|
opt-einsum
|
platformdirs
|
requests
|
scipy
|
toolz
|
typing-extensions
Optional dependencies:
gymnasium
|
hydra-core
|
omegaconf
|
opencv-python-headless
|
pygame
|
pytest
|
pytest-github-actions-annotate-failures
|
pytest-sugar
|
pytest-xdist
|
rtpt
|
safetensors
|
syrupy
|
tensorboard
|
torch
|
tyro
|
wandb
Downloads last day:
7
Downloads last week:
29
Downloads last month:
111
7-day smoothing