Alex Ziller

icon indicating an email [email protected]

icon location Munich icon location PostDoc at Technical University of Munich (Chair for AI in Medicine and Healthcare)

Results 4 issues of Alex Ziller

mean sum aggregation

subsampling amplification accounting

update jax.tree.map to comply with jax 0.4.34

4 comment

Pull Request to fix #1093

Jax 0.4.34 raises an error for None as tree-prefix

2 comment

Hello:) [The most recent release of jax 0.4.34](https://github.com/jax-ml/jax/releases/tag/jax-v0.4.34) does no longer support anything of the kind `jax.tree.map(f, None, non-None)`. This also affects optax, for example, in [add_decayed_weights](https://github.com/google-deepmind/optax/blob/main/optax/transforms/_adding.py#L50). The error can...