Skip to main content
Donate to the Python Software Foundation or Purchase a PyCharm License to Benefit the PSF! Donate Now

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.

Goals

HeAT is a flexible and seamless open-source software for high performance data analytics and machine learnings. It provides highly optimized algorithms and data structures for tensor computations using CPUs, GPUs and distributed cluster systems on top of MPI. The goal of HeAT is to fill the gap between data analytics and machine learning libraries with a strong focus on on single-node performance, and traditional high-performance computing (HPC). HeAT’s generic Python-first programming interface integrates seamlessly with the existing data science ecosystem and makes it as effortless as using numpy to write scalable scientific and data science applications.

HeAT allows you tackle your actual Big Data challenges that go beyond the computational and memory needs of your laptop and desktop.

Features

  • High-performance n-dimensional tensors
  • CPU, GPU and distributed computation using MPI
  • Powerful data analytics and machine learning methods
  • Abstracted communication via split tensors
  • Python API

Getting Started

Check out our Jupyter Notebook [tutorial](https://github.com/helmholtz-analytics/heat/blob/master/scripts/tutorial.ipynb) right here on Github or in the /scripts directory.

Requirements

HeAT is based on [PyTorch](https://pytorch.org/). Specifially, we are exploiting PyTorch’s support for GPUs and MPI parallelism. For MPI support we utilize [mpi4py](https://mpi4py.readthedocs.io). Both packages can be installed via pip or automatically using the setup.py.

Installation

Tagged releases are made available on the [Python Package Index (PyPI)](https://pypi.org/project/heat/). 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 https://github.com/helmholtz-analytics/heat.git

License

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

Acknowledgements

This work is supported by the [Helmholtz Association Initiative and Networking](https://www.helmholtz.de/en/about_us/the_association/initiating_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.

Filename, size & hash SHA256 hash help File type Python version Upload date
heat-0.1.0.tar.gz (53.3 kB) Copy SHA256 hash SHA256 Source None

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 SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page