Add TensorFlow array API backend by njzjz · Pull Request #441 · data-apis/array-api-compat · GitHub
Skip to content

Add TensorFlow array API backend#441

Open
njzjz wants to merge 4 commits into
data-apis:mainfrom
njzjz:add-tensorflow-backend
Open

Add TensorFlow array API backend#441
njzjz wants to merge 4 commits into
data-apis:mainfrom
njzjz:add-tensorflow-backend

Conversation

@njzjz

@njzjz njzjz commented Jun 27, 2026

Copy link
Copy Markdown

Summary

  • add a TensorFlow compatibility namespace with creation, elementwise, manipulation, linalg, FFT, dtype inspection, and namespace info support
  • wire TensorFlow into package metadata, install sources, helper detection, and tests
  • add TensorFlow array-api-tests skips/xfails files and CI workflow

Testing

  • PYTHONPATH=src python -m pytest -q
  • ARRAY_API_TESTS_MODULE=array_api_compat.tensorflow ARRAY_API_TESTS_VERSION=2025.12 ARRAY_API_TESTS_XFAIL_MARK=skip PYTHONPATH=/home/jzzeng/codes/array-api-compat/src python -m pytest -q /tmp/array-api-tests-master/array_api_tests --xfails-file /home/jzzeng/codes/array-api-compat/tensorflow-xfails.txt --skips-file /home/jzzeng/codes/array-api-compat/tensorflow-skips.txt --disable-deadline -n 2

Notes

  • TensorFlow backend avoids NumPy fallbacks and tf.experimental.numpy.
  • Remaining skips are limited to TensorFlow tensor object surface gaps and true-division dunder array-array promotion.

@lucascolley

Copy link
Copy Markdown
Member

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants