Array API Standard Support: optimize#
Support on CPU#
Legend
✔️ = supported
✖ = unsupported
N/A = out-of-scope
blank = not currently documented
function |
torch |
jax |
dask |
|---|---|---|---|
✔️ |
✔️ |
✔️ |
|
✔️ |
✖ |
✔️ |
|
✔️ |
✔️ |
✔️ |
|
✔️ |
✖ |
✔️ |
|
Support on GPU#
Legend
✔️ = supported
✖ = unsupported
N/A = out-of-scope
blank = not currently documented
function |
cupy |
torch |
jax |
|---|---|---|---|
✔️ |
✔️ |
✔️ |
|
✔️ |
✔️ |
✖ |
|
✔️ |
✔️ |
✔️ |
|
✔️ |
✔️ |
✖ |
|
Support with JIT#
Legend
✔️ = supported
✖ = unsupported
N/A = out-of-scope
blank = not currently documented
function |
jax |
|---|---|
✔️ |
|
✖ |
|
✔️ |
|
✖ |
|