Jonatan Kłosko

If there is a list of operations that are inherently not supported for jax-metal, it would be very helpful to have the list documented somewhere :)

Closing in favour of #2868.

(I pushed an update to reduce the diff, in case someone wants to look into another PjRt plugin integration)

There are quantized tensor types in [StableHLO spec](https://github.com/openxla/stablehlo/blob/main/docs/spec.md), but looks like there are still discussions around it, for example https://github.com/openxla/stablehlo/issues/1491. Also, that's only the specification, based on https://github.com/openxla/xla/issues/9291 I infer...

This is confusing, I cannot reproduce it using Docker. ``` $ docker run --rm -it --platform hexpm/elixir:1.16.3-erlang-26.2.5-ubuntu-jammy-20240427 $ apt update && apt install wget build-essential $ iex iex> Mix.install([:exla]) ```...

Ah, I was assuming that the glibc mismatch has to do with the XLA binary, but it may very well be the EXLA NIF, and that makes more sense. When...

@josevalim given that we cache globally, wouldn't it be enough to either copy from the global cache to _build or let it compile otherwise?

@josevalim we cache EXLA globally, #1016 :p

@christianjgreen I think you still want to set the env var beforehand, export it in your terminal session where you start `iex`/`livebook` (or in `~/.livebookdesktop.sh`, in case of Livebook Desktop).

Did you try building by setting the XLA revision as in https://github.com/elixir-nx/xla/issues/63#issuecomment-1844195261? Setting up the right environment for building was an issue before, that's why we have the Dockerfile. I...