Repository navigation

Array API compatibility for pytorch and jax - #91

Open

faymanns wants to merge 15 commits into

mainfrom

Open

Array API compatibility for pytorch and jax#91
faymanns wants to merge 15 commits into
mainfrom
array_api

Commits on Oct 6, 2026
- First working version of basis functions with array_api_compat
  faymanns
  committed
  3589097
- Fixed wrong name for pytorch
  faymanns
  committed
  5e0c079
- Stop pytest after first failed test
  faymanns
  committed
  f0c1cbb
- Pass environement variables for caching torch compiles
  faymanns
  committed
  0e39e26
- Add MSVC for windows runners
  faymanns
  committed
  c8ac18b
- Only add MSVC on windows
  faymanns
  committed
  7c331ed
- Cache compiled jax
  faymanns
  committed
  0d5475c
- Module-level caching to avoid recompiles for non-static methods such as
  faymanns
  committed
  9f08a00
- Cache compiled methods on class-level
  faymanns
  committed
  c6f4186
- spline __call__ working with pytorch
  faymanns
  committed
  818ba5e
Commits on Oct 11, 2026
- Test for different backends with fixtures
  faymanns
  committed
  c1d89d2
- Test output of basis function calls
  faymanns
  committed
  ee35966
Commits on Oct 12, 2026
- Test that output of basis function calls has the right dtype
  faymanns
  committed
  9db3eaa
- Test that output of basis function filters has the right dtype
  faymanns
  committed
  da9a802
Commits on Oct 13, 2026
- Remove prints
  faymanns
  committed
  165d8f9