←BackDdeepmind/distrax0Copy as MarkdownView on GitHub↗0 stars·0 forks·0 viewsDistraxFeaturesJAX Frameworks - Lightweight library of probability distributions and JAX-native bijectors.