#

distrax

Here are 2 public repositories matching this topic...

frankroeder / goal_conditioned_rl

Goal-conditioned reinforcement learning like 🔥

goals reinforcement-learning robotics deep-reinforcement-learning flax gymnasium actor-critic deep-deterministic-policy-gradient jax soft-actor-critic optax goal-conditioned-rl distrax gymnasium-robotics mpi4jax droq

Updated Feb 3, 2024
Python

Raffaelbdl / distrax_flax

Tabulate Flax modules that use distrax Distributions

flax distrax

Updated Dec 15, 2023
Python

Improve this page

Add a description, image, and links to the distrax topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the distrax topic, visit your repo's landing page and select "manage topics."