heat
heat copied to clipboard
Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
## Description The reuse of gshape array as the receiving buffer made the code a bit too difficult to understand. So, created a new array `neighbour_shape` to hold, well...., the...
Array API
## Description Implement array API according to the [Python array API standard (version 2021.12)](https://data-apis.org/array-api/2021.12/) See #774 for more information. ## Changes proposed: - In conformance with the specification, the new...
## Description Implement `broadcast_arrays`, `broadcast_to` Issue/s resolved: #778 ## Changes proposed: - ## Type of change - New feature (non-breaking change which adds functionality) ## Due Diligence - [ ]...
## Description Implemented 2D convolution with distributed kernel support. Issue resolved: #920 skip ci
## Description Bulge Chasing algorithm to transform the given matrix into a bidiagonal matrix (Upper bidiagonal in case of m>=n and lower bidiagonal in case of m ## Changes proposed:...
## Description - Fixes a bug where adding a dimension (via slicing with `None`) on an array with an axis of length 0 results in an error. - Fixes a...
## Description Renames `keepdim` kwarg to `keepdims`, in accordance with the numpy API. Affected functions: - `prod` - `sum` - `all` - `any` - `max` - `median` - `min` -...
## Description ### Distributed Compressed Sparse Row Matrix: Dcsr_matrix A format for the efficient storage and manipulation of sparse data (with majority 0s). This distributed implementation builds upon the [torch.sparse_csr_tensor](https://pytorch.org/docs/stable/generated/torch.sparse_csr_tensor.html)...
See https://docs.xarray.dev/en/stable/ If I understand correctly, an `xarray` object is made up of the actual `data` array (np.ndarray), and ~~1-D~~ `coordinates` arrays (dictionaries?) that map `data` dimensions and indices to...
### What happened? Special versions of pytorch adds a plus sign and some text to the version number like `1.12.1+cpu` or `1.12.1+cu116`. When installing Heat, these packages get replaced (`1.12.1`)....