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.
I am working on adding support for pytorch. A good basis for this work is @pmeier's wrapper code at data-apis/array-api-tests@master...pmeier:array-api-tests:pytorch, and also @AnirudhDagar scipy pytorch demo scipy/scipy@main...AnirudhDagar:scipy:array-api-demo.