dodox/itpal_jax
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-12-21 18:53:44 +01:00
parent e83cb9a8a5
commit 2e0ca977bc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -12,7 +12,7 @@ JAX bindings and native implementations of differentiable trust region projectio
- Multiple projection types: - Multiple projection types:
- KL (Kullback-Leibler divergence) - KL (Kullback-Leibler divergence)
- Wasserstein (only diagonal covariance) - Wasserstein (only diagonal covariance)
- Frobenius (wip, not tested) - Frobenius (wip, problem with cov projections)
- Identity (no projection) - Identity (no projection)
- Support for both diagonal and full covariance Gaussians (induced from cholesky decomposition) - 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) - Contextual and non-contextual standard deviations (non-contextual means all standard deviations in batch are expected to be the same)