FlowJAX

https://github.com/danielward27/flowjax

FlowJAX provides distributions, bijections, and normalizing flows using Equinox and JAX, featuring a...

Tokens:15,758
Snippets:58
Trust Score:7.4
License:MIT
Update:9 months ago
Tokens:
Raw