PyPI page
Home page
Author:
None
License:
MIT License Copyright (c) 2024 Ryan Strauss Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal...
Summary:
Reinforcement learning in pure JAX.
Latest version:
0.2.1
Required dependencies:
brax
|
chex
|
click
|
distrax
|
dm-env
|
dm-haiku
|
einops
|
ffmpeg
|
flashbax
|
gymnax
|
imageio
|
jax
|
jaxlib
|
mctx
|
ml-collections
|
moviepy
|
numpy
|
optax
|
pgx
|
rlax
|
tqdm
|
wandb
Downloads last day:
0
Downloads last week:
10
Downloads last month:
26