Implementation of Reinforcement Learning agents in JAX
Project description
Jaxagents
Jaxagents is a Python implementation of Reinforcement Learning agents built upon JAX.
Content
So far, the project includes the following agents:
- Q-learning:
- Deep Q Networks (DQN)
- Double Deep Q Networks (DDQN)
- Categorical Deep Q Networks (often known as C51)
- Quantile Regression Deep Q Networks (QRDQN)
- Policy gradient:
- REINFORCE
- PPO with clipping and GAE
Background
Research and development in Reinforcement Learning can be computationally cumbersome. Utilizing JAX's high computational performance, Jaxagents provides a framework for applying and developing Reinforcement Learning agents that offers benefits in:
- computational speed
- easy control of random number generation
- hyperparameter optimization (via parallelized calculations)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
jaxagents-0.1.7.tar.gz
(34.4 kB
view hashes)
Built Distribution
jaxagents-0.1.7-py3-none-any.whl
(39.1 kB
view hashes)
Close
Hashes for jaxagents-0.1.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | da42da9a5d52adc8d3603a3d582c87f271fab59eb5733f474dff1a07532cf883 |
|
MD5 | ce5d4e4bdd24442c6275de54d818f7c3 |
|
BLAKE2b-256 | 749bca4188faab7d9f70325a3f48d338981305c5c9f32b7227af21c9f7ca2259 |