John Stewart

Nx.Random.uniform(key) error when using mps backend in Torchx.

4

Calling `Nx.Random.uniform(key)` throws an error when using mps backend in Torchx. ```elixir Mix.install( [ {:nx, "~> 0.9.2"}, {:torchx, "~> 0.9.2"} ] ) :erlang.system_flag(:backtrace_depth, 100) Nx.default_backend({Torchx.Backend, device: :mps}) key = Nx.Random.key(42)...