Julius Kunze

icon indicating a website link https://juliuskunze.com icon indicating an email [email protected]

icon location Berlin icon location Let's create helpful intelligent machines.

Results 2 issues of Julius Kunze

Fast custom division by 0

3 comment

Is there an efficient GPU (i. e. XLA one-op) implementation of floating-point division available in JAX where division by `0` returns `0`, instead of ~raising an error~ `inf`? In my...

question
P2 (eventual)
NVIDIA GPU

Cache miss breaks jit(parametrized(jit).apply)

See `TODO` in `test_parametrized_jit`.