Lena Martens

Add more documentation on calling JAX transforms inside a Haiku module.

3

We have https://dm-haiku.readthedocs.io/en/latest/notebooks/transforms.html, but I think we can add some more context on what error you get (an UnexpectedTracerError) and some alternatives to hk.lift (eg. if the only side-effect is...

checks as XLARuntimeErrors! no need to discharge! I should probably add more tests here but sending for early review for `emit_python_callback` usage.

- `err.throw` == `check_error(err)`: meaning they now have the same behavior under checkify (the error is discharged even when running an eager checkify) - `divided by zero` -> `division by...

Raise cond index into tracing context in case of effects

1

So even if the cond is not data dependent at all, it's included in the dynamic trace. Fixes #12385. But .. is it safe to do this? Not sure! (also...

Compilation failing with "RuntimeError: Cannot open compilation feedback"

3

## Summary See the attached stack-trace below: compilation fails at the point of collecting compilation feedback with a message of ` [6:21, byte=99]: Expected object key`, and dumps the compilation-feedback...

Build dex without CUDA support, then run `dex --backend llvm-cuda repl` Throws: ``` dex: Missing symbol: MangledSymbol "dex_ensure_has_cuda_context" CallStack (from HasCallStack): error, called at src/lib/LLVM/V9/JIT.hs:105:18 in dex-0.1.0.0-3TQe6uju44BJFUXpaseRHY:LLVM.V9.JIT ``` (or sometimes...