diff --git a/README.md b/README.md index 89204ea..1a74b5d 100644 --- a/README.md +++ b/README.md @@ -12,7 +12,7 @@ JAX bindings and native implementations of differentiable trust region projectio - Multiple projection types: - KL (Kullback-Leibler divergence) - Wasserstein (only diagonal covariance) - - Frobenius (wip, not tested) + - Frobenius (wip, problem with cov projections) - Identity (no projection) - Support for both diagonal and full covariance Gaussians (induced from cholesky decomposition) - Contextual and non-contextual standard deviations (non-contextual means all standard deviations in batch are expected to be the same)