Farama-Foundation / Farama-Foundation/Jumpy

Add `jumpy.random.Generator`

Open
#27 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
56
Forks
9
PR merge metrics
No merged PRs in 30d

Description

Jumpy's random functions are a bit weird for numpy as it uses a `np.array` as the key but needs to convert this to the `RandomState` for each function call.
Imo, as Gymnasium and other projects use `np.random.Generator` then it makes it difficult to easily change to using jumpy rather than numpy.

Therefore, the proposal is to create a `jp.random.Generator` object that is equivalent to `np.random.Generator` for numpy and for jumpy, behind the scenes we store the key and split the `prng` and return the results.

To consider is if the object inherits from `np.random.Generator` with functions being overrides with a check if numpy or jax is used.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.