alenax
A JAX wrapper for the Arcade Learning Environment, compatible with Gymnax.
Installation
pip install alenax
Requires Python >= 3.10.
Usage
import jax
from alenax import AtariEnv
env = AtariEnv("pong")
key = jax.random.PRNGKey(0)
obs, state = env.reset(key)
obs, state, reward, done, info = env.step(key, state, action)
All environment methods are JIT-compiled and support jax.vmap for batched execution.
Wrappers
from alenax import ClipReward, EpisodicLife, RecordEpisodeStatistics
env = AtariEnv("pong")
env = ClipReward(env) # Clip rewards to [-1, 1]
env = EpisodicLife(env) # Treat life loss as episode end
env = RecordEpisodeStatistics(env) # Track episode returns and lengths
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
alenax-0.1.3.tar.gz
(5.2 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file alenax-0.1.3.tar.gz.
File metadata
- Download URL: alenax-0.1.3.tar.gz
- Upload date:
- Size: 5.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
uv/0.9.18 {"installer":{"name":"uv","version":"0.9.18","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
61bbdfdd877530464f09705ace1b84308434f444db4307ed5bdba373dd2b4dda
|
|
| MD5 |
eb5422222b8a0ed70cfe0abf023d2f0a
|
|
| BLAKE2b-256 |
5f612862b03518989ccef85e5c853ea8bc72cd094762d68d94fcc0651d3a5373
|
File details
Details for the file alenax-0.1.3-py3-none-any.whl.
File metadata
- Download URL: alenax-0.1.3-py3-none-any.whl
- Upload date:
- Size: 5.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
uv/0.9.18 {"installer":{"name":"uv","version":"0.9.18","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
80914590537adad8f0c823799147dea6c57ae4bf2723b2859a661bb5045288d7
|
|
| MD5 |
4f14caeb0072a8da6726ced8218f3d15
|
|
| BLAKE2b-256 |
4b2664b395d6cf820b2021a9d180721610e971d0a55bed35aae5c61dc5a56cde
|
