Skip to main content

A framework for high performance data analytics and machine learning.

Project description

![HeAT Logo](doc/images/logo_HeAT.png)

HeAT is a distributed tensor framework for high performance data analytics.


The goal of HeAT is to fill the gap between machine learning libraries that have a strong focus on exploiting GPUs for performance, and traditional, distributed high-performance computing (HPC). The basic idea is to provide a generic, distributed tensor library with machine learning methods based on it.

Among other things, the implementation will allow us to tackle use cases that would otherwise exceed memory limits of a single node.


  • high-performance n-dimensional tensors
  • CPU, GPU and distributed computation using MPI
  • powerful machine learning methods using above mentioned tensors


HeAT is based on [PyTorch]( Specifially, we are exploiting PyTorch’s support for GPUs and MPI parallelism. Therefore, PyTorch must be compiled with MPI support when using HeAT. The instructions to install PyTorch in that way are contained in the script [](, which we’re also using to install PyTorch in Travis CI.


Tagged releases are made available on the [Python Package Index (PyPI)]( You can typically install the latest version with

> $ pip install heat

If you want to work with the development version, you can checkout the sources using

> $ git clone


HeAT is distributed under the MIT license, see our [LICENSE](LICENSE) file.


This work is supported by the [Helmholtz Association Initiative and Networking]( Fund under project number ZT-I-0003.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for heat, version 0.0.2
Filename, size File type Python version Upload date Hashes
Filename, size heat-0.0.2.tar.gz (2.0 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page