distrax

There are 3 repositories under distrax topic.

frankroeder/goal_conditioned_rl
Goal-conditioned reinforcement learning like 🔥
Language:Python6 2 10
AI-Ahmed/ppcax
PPCAx: Stochastic dimensionality reduction with JAX-ecosystem
Language:Python
Raffaelbdl/distrax_flax
Tabulate Flax modules that use distrax Distributions
Language:Python1 0