Use `api-array-compat` as a backend

MLResearchAtOSRAM / tmm_fast

tmm_fast is a lightweight package to speed up optical planar multilayer thin-film device computation. Developed by Alexander Luce (@Nerrror) in cooperation with Heribert Wankerl (@HarryTheBird).

MIT License

53 stars 22 forks source link

Use `api-array-compat` as a backend #22

Open kadykov opened 1 year ago

kadykov commented 1 year ago

What do you think about the api-array-compat project?

It should be not so difficult to change the backend from PyTorch to api-array-compat, and it will give many advantages:

Simplify array type selection/verification logic
Add cupy support for those who want to calculate on GPU, but don't need backpropagation or don't want to install PyTorch
Make PyTorch dependence optional