inducer / pytato

Lazily evaluated arrays in Python
Other
9 stars 16 forks source link
array arraylang code-generation code-transformation dag data-flow-graph numpy

Pytato: Get Descriptions of Array Computations via Lazy Evaluation

.. image:: https://gitlab.tiker.net/inducer/pytato/badges/main/pipeline.svg :alt: Gitlab Build Status :target: https://gitlab.tiker.net/inducer/pytato/commits/main .. image:: https://github.com/inducer/pytato/workflows/CI/badge.svg?branch=main :alt: Github Build Status :target: https://github.com/inducer/pytato/actions?query=branch%3Amain+workflow%3ACI .. image:: https://badge.fury.io/py/pytato.png :alt: Python Package Index Release Page :target: https://pypi.org/project/pytato/

Imagine TensorFlow, but aimed at HPC. Produces a data flow graph, where the edges carry arrays and the nodes are (give or take) static-control programs that compute array outputs from inputs, possibly (but not necessarily) expressed in Loopy <https://github.com/inducer/loopy>__. A core assumption is that the graph represents a computation that's being repeated often enough that it is worthwhile to do expensive processing on it (code generation, fusion, OpenCL compilation, etc).

Pytato is licensed to you under the MIT/X Consortium license. See the documentation <https://documen.tician.de/pytato/misc.html>__ for further details.

Numpy compatibility

Pytato is written to pose no particular restrictions on the version of numpy used for execution. To use mypy-based type checking on Pytato itself or packages using Pytato, numpy 1.20 or newer is required, due to the typing-based changes to numpy in that release.

Furthermore, pytato now uses type promotion rules based on those in numpy <https://numpy.org/devdocs/numpy_2_0_migration_guide.html#changes-to-numpy-data-type-promotion>__ that should result in the same data types as the currently installed version of numpy.