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.
Add a description, image, and links to the ipu topic page so that developers can more easily learn about it.
To associate your repository with the ipu topic, visit your repo's landing page and select "manage topics."