The Ocean Tensor Package

Ewout van den Berg

doi:10.5334/jors.268

The Ocean Tensor Package

Journal of Open Research Software

Volume 7 (2019): Issue 1

By: Ewout van den Berg

Open Access

|Aug 2019

Abstract

The Ocean Tensor Package is an open-source package for matrix and tensor operations on CPU and GPU. The package aims to serve as a foundational layer for applications that require dense tensor operations on a variety of device types. All operations are available through a unified interface that is carefully designed to be powerful, extensible, and at the same time easy to use. The package has a modular implementation in C and provides a light-weight Python interface. Modularity of the package facilitates the addition of new operations as well as new device types.

References

Abadi, M, Agarwal, A, Barham, P, Brevdo, E, Chen, Z, Citro, C, Corrado, G S, Davis, A, Dean, J, Devin, M, Ghemawat, S, Goodfellow, I, Harp, A, Irving, G, Isard, M, Jia, Y, Jozefowicz, R, Kaiser, L, Kudlur, M, Levenberg, J, Mané, D, Monga, R, Moore, S, Murray, D, Olah, C, Schuster, M, Shlens, J, Steiner, B, Sutskever, I, Talwar, K, Tucker, P, Vanhoucke, V, Vasudevan, V, Viégas, F, Vinyals, O, Warden, P, Wattenberg, M, Wicke, M, Yu, Y and Zheng, X 2015 TensorFlow: Large-scale machine learning on heterogeneous systems. Software available from tensorflow.org.
Search in Google Scholar Back to article
Broekema, P C, Mol, J J D, Nijboer, R, van Amesfoort, A S, Brentjens, M A, Loose, G M, Klijn, W F A and Romein, J W 2018 A GPU-based correlator and beamformer for LOFAR. arXiv: 1801.04834. DOI: 10.1016/j.ascom.2018.04.006
Open DOI Search in Google Scholar Back to article
Cano, A 2018 A survey on graphic processing unit computating for large-scale data mining. WIREs Data Mining Knowledge discovery, 8: e1232. DOI: 10.1002/widm.1232
Open DOI Search in Google Scholar Back to article
Chen, T, Li, M, Li, Y, Lin, M, Wang, N, Wang, M, Xiao, T, Xu, B, Zhang, C and Zhang, Z 2015 MXNet: A flexible and efficient machine learning library for heterogeneous distributed systems. In Neural Information Processing Systems, Workshop on Machine Learning Systems.
Search in Google Scholar Back to article
Collobert, R, Bengio, S and Marithoz, J 2002 Torch: A modular machine learning software library.
Search in Google Scholar Back to article
Collobert, R, Kavukcuoglu, K and Farabet, C 2011 Torch7: A Matlab-like environment for machine learning. In BigLearn, NIPS Workshop, number EPFLCONF-192376.
Search in Google Scholar Back to article
Jia, Y, Shelhamer, E, Donahue, J, Karayev, S, Long, J, Girshick, R, Guadarrama, S and Darrell, T 2014 Caffe: Convolutional architecture for fast feature embedding. In Proceedings of the 22nd ACM International Conference on Multimedia, 675–678. DOI: 10.1145/2647868.2654889
Open DOI Search in Google Scholar Back to article
Kutzner, C, Páll, S, Fechner, M, Esztermann, A, de Groot, B L and Grubmüller, H 2015 Best bang for your buck: GPU nodes for GROMACS biomolecular simulations. Journal of Computational Chemistry, 36(26): 1990–2008. DOI: 10.1002/jcc.24030
Open DOI Search in Google Scholar Back to article
NVIDIA. GPU-accelerated applications. https://www.nvidia.com/en-us/data-center/gpu-accelerated-applications/catalog/.
Search in Google Scholar Back to article
Okuta, R, Unno, Y, Nishino, D, Hido, S and Loomis, C 2017 CuPy: A NumPycompatible library for NVIDIA GPU calculations. In Proceedings of Workshop on Machine Learning Systems (LearningSys) in The Thirty-first Annual Conference on Neural Information Processing Systems (NIPS).
Search in Google Scholar Back to article
Oliphant, T E 2006 A guide to Numpy. Trelgol Publishing.
Search in Google Scholar Back to article
Paszke, A, Gross, S, Chintala, S, Chanan, G, Yang, E, DeVito, Z, Lin, Z, Desmaison, A, Antiga, L and Lerer, A 2017 Automatic differentiation in PyTorch. In NIPS 2017 Autodiff Workshop.
Search in Google Scholar Back to article
Shi, L, Liu, W, Zhang, H, Xie, Y and Wang, D 2012 A survey of GPU-based medical image computing techniques. Quantitative Imaging in Medicine and Surgery, 2(3): 188–206.
Search in Google Scholar Back to article
Shi, X, Zheng, Z, Zhou, Y, Jin, H, He, L, Liu, B and Hua, Q-S 2018 Graph processing on GPUs: A survey. ACM Computing Surveys, 50(6): 81: 1–35. DOI: 10.1145/3128571
Open DOI Search in Google Scholar Back to article
Trefethen, L N and Bau, D, III 1997 Numerical Linear Algebra. SIAM. DOI: 10.1137/1.9780898719574
Open DOI Search in Google Scholar Back to article
van den Berg, E 2018 The Ocean Tensor Package. Available at https://github.com/ibm/ocean-tensor-package.
Search in Google Scholar Back to article
Wang, H, Peng, H, Chang, Y and Liang, D 2018 A survey of GPU-based acceleration techniques in mri reconstructions. Quantitative Imaging in Medicine and Surgery, 8(2): 196–208. DOI: 10.21037/qims.2018.03.07
Open DOI Search in Google Scholar Back to article
Yalamanchili, P, Arshad, U, Mohammed, Z, Garigipati, P, Entschev, P, Kloppenborg, B, James, J and Melonakos, J 2015 ArrayFire – a high performance software library for parallel computing with an easy-to-use API. https://github.com/arrayfire/arrayfire.
Search in Google Scholar Back to article