distrax
There are 3 repositories under distrax topic.
frankroeder/goal_conditioned_rl
Goal-conditioned reinforcement learning like 🔥
AI-Ahmed/ppcax
PPCAx: Stochastic dimensionality reduction with JAX-ecosystem
Raffaelbdl/distrax_flax
Tabulate Flax modules that use distrax Distributions