Skip to content

Latest commit

 

History

History
36 lines (27 loc) · 1.07 KB

README.md

File metadata and controls

36 lines (27 loc) · 1.07 KB

PyKokkos

PyKokkos is a framework for writing high-performance Python code similar to Numba. In contrast to Numba, PyKokkos kernels are primarily parallel and are also performance portable, meaning that they can run efficiently on different hardware (CPUs, NVIDIA GPUs, and AMD GPUs) with no changes required.

For more information about PyKokkos, see the PyKokkos GitHub pages: https://kokkos.github.io/pykokkos/index.html

Installation

Please follow the instructions at: https://kokkos.github.io/pykokkos/installation.html

Citation

If you have used PyKokkos in a research project, please cite this research paper:

@inproceedings{AlAwarETAL21PyKokkos,
  author = {Al Awar, Nader and Zhu, Steven and Biros, George and Gligoric, Milos},
  title = {A Performance Portability Framework for Python},
  booktitle = {International Conference on Supercomputing},
  pages = {467-478},
  year = {2021},
}

Acknowledgments

This project is partially funded by the U.S. Department of Energy, National Nuclear Security Administration under Award Number DE-NA0003969 (PSAAP III).