Versions

Description

augpy is a lightweight library with minimal dependencies that provides a comprehensive tensor implementation for Cuda-enabled GPUs, with most of the functionality you are used to from numpy and Pytorch with a similar syntax. What sets augpy apart is its focus on saturating math (no under or overflows possible), as well as comprehensive support for all data types in all functions. For example, augpy allows you to fill a uint8 tensor with Gaussian distributed random numbers. augpy's tensors are based on the DLPack specification and can be exchanged copy-free with other frameworks such as Jax or Pytorch. While augpy's support for tensors is quite generic, it also includes additional functionality to work on 2D images, such as high quality affine warps with supersampling and Gaussian blurs.

Repository

https://gitlab.com/jfolz/augpy.git

Project Slug

augpy

Last Built

4 weeks, 1 day ago passed

Maintainers

Home Page

https://gitlab.com/jfolz/augpy

Badge

Tags

python, cuda, image, cpp, tensor, data-augmentation, jpeg

Short URLs

augpy.readthedocs.io
augpy.rtfd.io

Default Version

latest

'latest' Version

master