jax-metal for jax.random.PRNGKey generation

Machine Learning & AI General

krisyang OP

Created Jun ’24

Replies 0

Boosts 0

Participants 1

Hi all, I couldn't use random.PRNGKey to generate random seed. Wondering anyone has similar issue before and figure this out. Here is my current config: jax-metal==0.0.3, jaxlib==0.4.10, jax==0.4.11.

I am using Apple M1 pro.

Boost