-
## 🐛 Bug
My training run with PyTorch XLA is running very slowly (on TPU v3-64), and as suggested by the debug messages, I am submitting dumps of the execution graphs and soliciting feedback on how…
-
### Description
When I attempt to build jaxlib v0.4.30 on a Jetson Orin with the command `python build/build.py --enable_cuda --cuda_path /usr/local/cuda-12.2 --cudnn_path /usr/lib/aarch64-linux-gnu …
-
cutlass is used in building kernels in tensorflow
I took a look in cutlass_archive/include/cutlass/matrix.h and indeed set_slice3x3 is not defined however set_slice_3x3 is
did not want to submi…
-
When running the example in the README, it crashes with the following error.
```
julia> f=Reactant.compile(sinsum_add, (input1,input2))
F0626 06:23:06.397895 67363613 simple_orc_jit.cc:311] Check…
-
### Issue type
Build/Install
### Have you reproduced the bug with TensorFlow Nightly?
No
### Source
source
### TensorFlow version
2.18.0
### Custom code
No
### OS platform and distribution
…
-
I see in the documentation of MuJoCO XLA (https://mujoco.readthedocs.io/en/stable/mjx.html) that sensors are under development, any chance this feature has been added? If so, are there any examples?
-
I installed Keypoint-Moseq using `pip` into my folder on the HPC (could not get the `conda` method of installation to work, was unable to find `_cuda`) and I have this error:
`2024-07-16 13:54:47.3…
-
I am trying to analyze the optimized HLO outputs for a few `Nx` functions using this flag in a live book: `system_env: %{"XLA_FLAGS" => "--xla_dump_to=/tmp/hlo"}`, however no dumps are generated.
…
-
### Description of the bug:
When I convert object detection with nms ops and filter ops, has below error:
RuntimeError: This model contains ops not capturable by Pytorch/XLA: aten::nonzero=6
### …
-
I'm currently building a library in Rust that leverages XLA and PJRT and while going through the APIs and the implementation of the Python bindings, I'm a little confused about the terminology and the…