Skip to main content

Jaxagents

Jaxagents is a Python implementation of Reinforcement Learning agents built upon JAX.

Content

So far, the project includes the following agents for Reinforcement Learning:

  • 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

So far, the project includes the following agents for Multi-Agent Reinforcement Learning:

  • Policy gradient:
    • IPPO

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)

Metadata

Release files for jaxagents 0.2.5

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for jaxagents 0.2.5
File Size Uploaded Provenance Actions
jaxagents-0.2.5.tar.gz 52.5 kB Without provenance Details for jaxagents-0.2.5.tar.gz

Built distribution (wheel)

Table of built distributions (wheels) for jaxagents 0.2.5
File Interpreter ABI Platform Actions
jaxagents-0.2.5-py3-none-any.whl Python 3 none any Details for jaxagents-0.2.5-py3-none-any.whl

Total release size: 115.1 kB

Release files / jaxagents-0.2.5.tar.gz

Download URL jaxagents-0.2.5.tar.gz
Size 52.5 kB
Tags Source
SHA-256 checksum
How to use checksums
628f46862c890c0b7c01aaf9e01a3228f775c2348fde5200da9f91897eea0adf
BLAKE2b-256 checksum
How to use checksums
753d2b4b0c116037712fa637aef3724f7625b1f195da6c348f8b4bba23fc2dae
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via poetry/2.1.1 CPython/3.12.7 Windows/11

Release files / jaxagents-0.2.5-py3-none-any.whl

Download URL jaxagents-0.2.5-py3-none-any.whl
Size 62.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
92299b2670c62bd6c1023e57a4c67323023064f873cf0c2bb3a4b39ca9012745
BLAKE2b-256 checksum
How to use checksums
e74bfba8390abe8b2af1631f4d833dc22664e3569d5e92452c0ce27ec39eecd8
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via poetry/2.1.1 CPython/3.12.7 Windows/11

Security for jaxagents 0.2.5

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page