JAX for Graphcore IPU (experimental)
-
Updated
Mar 12, 2024 - Python
JAX for Graphcore IPU (experimental)
TessellateIPU: low level Poplar tile programming from Python
Python implementation of Iterative Proportional Updating (IPU). The IPU algorithm is a general case of iterative proportional fitting that can satisfy two disparate sets of marginals that do not agree on a single total.
To associate your repository with the ipu topic, visit your repo's landing page and select "manage topics."