Initial stable release

JAX-based normalizing flows and bijections library, with special focus on physics research. Features continuous normalizing flows, matrix Lie group operations, and flexible architecture blocks. Built on Flax NNX with focus on research flexibility and composability.