-
This issue is essentially a cupy version of #12398 that I am raising as a result of needing #12883 to update cudf for supporting recent changes to cupy. We currently claim to support cupy 9.5-11.*, bu…
vyasr updated
8 months ago
-
The spec for `to_device()` [says](https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.to_device.html#array_api.array.to_device):
> Copy the array from the device on w…
-
### Description
As in the title.
Reproducer:
```python
>>> import jax
>>> jax.config.update('jax_platform_name', 'cpu')
>>> a = jax.numpy.array([1, 2])
>>> hasattr(a, '__cuda_array_interfac…
pearu updated
9 months ago
-
`isort` does have some wacky heuristics to determine first v third party, but ultimately I'd love to elide the import-sorting it does with `ruff` :zap:
To me, doesn't have to be 1:1, so long as th…
-
Hello Jax developers - many thanks for the great package! Possible bug below ...
In NumPy, it's possible to view a complex array as a float array where the last dimension has twice the length. In J…
izaid updated
8 months ago
-
This is a tracking issue for tasks related to the 2023 revision of the array API specification. The issue is comprised of 3 parts: general todos, new APIs, and issues needing triage.
## TODO
- […
-
## 🐛 Bug
mlc_llm reports unexpective error "The model weight size may be larger than GPU memory size.", but the fact is that there is enough memory. I met with the same problem when trying to run …
-
## 🐛 Bug
crash log
```
libc++abi: terminating due to uncaught exception of type tvm::runtime::InternalError: [13:57:25] /Users/xxx/project/mlc-llm/3rdparty/tvm/src/runtime/relax_vm/lm_support.cc:…
-
Hello,
I'm trying to add some more augmentations not yet available in DALI using DLTensorPythonFunction.
I made some tests with tfa.image.mean_filter2d on cpu and the speed is more than 10 lower t…
-
MXNet imperative operator invocation overhead is as large as 30-60us, which is significant compared to the official NumPy operators with ~600ns overhead. This has negatively impacted the performance o…