NormalizingFlowFilters

This module provides an interface to use conditional normalizing flow for data assimilation.

• prior_state has shape (s..., N) for state shape s and number of ensemble members N.
• prior_obs has shape (r..., N) for observation shape r and number of ensemble members N.
• y_obs has shape r for observation shape r.