JAX

https://docs.jax.dev/en/latest/index.html
Admin
JAX is a Python library for accelerator-oriented array computation and program transformation,...
Tokens:2,945
Snippets:71
Trust Score:6.4
Update:6 months ago
Show doc for...
Tokens:
Raw
### JAX Experimental Sparse Operations

Source: https://docs.jax.dev/en/latest/index.html

This section covers experimental functions for handling sparse arrays in JAX, including slicing, updating, and conversions.

```APIDOC
## jax.experimental.sparse Operations

### Description
Provides experimental functionalities for manipulating sparse arrays.

### Methods
- `bcoo_dynamic_slice`
- `bcoo_extract`
- `bcoo_fromdense`
- `bcoo_gather`
- `bcoo_multiply_dense`
- `bcoo_multiply_sparse`
- `bcoo_update_layout`
- `bcoo_reduce_sum`
- `bcoo_reshape`
- `bcoo_slice`
- `bcoo_sort_indices`
- `bcoo_squeeze`
- `bcoo_sum_duplicates`
- `bcoo_todense`
- `bcoo_transpose

### Parameters
(Specific parameters for each function are not detailed in the provided text.)

### Request Body
(Not applicable for these function calls.)

### Response
(Specific return types for each function are not detailed in the provided text.)
```

--------------------------------

### JAX Array Methods

Source: https://docs.jax.dev/en/latest/index.html

This section lists common methods available on JAX arrays for various operations like aggregation, manipulation, and inspection.

```APIDOC
## jax.Array Methods

### Description
Common methods available on JAX arrays for data manipulation and analysis.

### Methods
- `addressable_shards`
- `all`
- `any`
- `argmax`
- `argmin`
- `argpartition`
- `argsort`
- `astype`
- `at`
- `byteswap`
- `choose`
- `clip`
- `compress`
- `committed`
- `conj`
- `conjugate`
- `copy`
- `copy_to_host_async`
- `cumprod`
- `cumsum`
- `device`
- `diagonal`
- `dot`
- `dtype`
- `flat`
- `flatten`
- `global_shards`
- `imag`
- `is_fully_addressable`
- `is_fully_replicated`
- `item`
- `itemsize`
- `max`
- `mean`
- `min`
- `nbytes`
- `ndim`
- `nonzero`
- `prod`
- `ptp`
- `ravel`
- `real`
- `repeat`
- `reshape`
- `round`
- `searchsorted`
- `shape`
- `sharding`
- `size`
- `sort`
- `squeeze`
- `std`
- `sum`
- `swapaxes`
- `take`
- `to_device`
- `trace`
- `transpose`
- `var`
- `view`
- `T` (alias for transpose)
- `mT` (alias for transpose)

### Parameters
(Specific parameters for each method are not detailed in the provided text.)

### Request Body
(Not applicable for these method calls.)

### Response
(Specific return types for each method are not detailed in the provided text.)
```
Tokens:
Raw