Johannes Brachem

icon indicating a website link https://www.jobrachem.com

icon company Georg-August-University Göttingen icon location Göttingen, Germany icon location Statistics PhD Student in Göttingen, Germany

Results 52 issues of Johannes Brachem

Replace Python loops in LieselInterface with native Jax

I keep experiencing relatively long compile times. I suspect, the Python loops in `LieselInterface.update_state`, and possibly in `LieselInterface.extract_position` might be partly responsible for this, see [this point in the Jax...

enhancement
comp:goose

jax version of GeneralizedExtremeValue distribution has wrong support for concentration=0

It seems like the support of the GEV is `(-1.0, inf)` for `concentration=0`, rather than the expected `(-inf, inf)`. I am using: - `tensorflow-probability 0.24.0` - `jax 0.4.30` I only...