felupe icon indicating copy to clipboard operation
felupe copied to clipboard
verified publisher icon adtzlr

Choosable array-backend

Open adtzlr opened this issue 2 years ago • 1 comments

Motivation

Easily switch between numpy, jax, torch or even cupy.

Usage

felupe.set_backend(numpy)

adtzlr avatar Oct 15 '23 16:10 adtzlr

SmallPebble has a nice implementation: https://github.com/sradc/SmallPebble/blob/main/smallpebble/array_library.py

adtzlr avatar Oct 17 '23 09:10 adtzlr