search

data-apis / array-api-compat

Compatibility layer for common array libraries to support the Array API
https://data-apis.org/array-api-compat/
MIT License
69 stars 22 forks source link

Pytorch support #9

Closed asmeurer closed 1 year ago

asmeurer commented 1 year ago

I am working on adding support for pytorch. A good basis for this work is @pmeier's wrapper code at https://github.com/data-apis/array-api-tests/compare/master...pmeier:array-api-tests:pytorch, and also @AnirudhDagar scipy pytorch demo https://github.com/scipy/scipy/compare/main...AnirudhDagar:scipy:array-api-demo.

pmeier commented 1 year ago

Keep in mind I haven't updated my fork in ~6 months.

rgommers commented 1 year ago

This was fixed by gh-14. PyTorch support is well-exercised by now.