Skip to content

Add JAX and NumPy examples#431

Open
danieldk wants to merge 27 commits intomainfrom
jax-support
Open

Add JAX and NumPy examples#431
danieldk wants to merge 27 commits intomainfrom
jax-support

Conversation

@danieldk
Copy link
Copy Markdown
Member

@danieldk danieldk commented Apr 8, 2026

This change adds JAX and NumPy examples, adds JAX +jax-tvm-ffi to dev/test shells, and adds some helper functions.

Fixes #402.

danieldk added 27 commits April 1, 2026 07:20
Before this change we erroneously inherited `callPackage` from nixpkgs.
This `callPackage` does not have access to the new scope (the ROCm/XPU
package set). However, this error was masked by `makeScope` overriding
`callPackage`, so `final` had the correct version. The latest nixpkgs
supports overriding `callPackage`, which surfaced this bug.
This check takes a long time, sometimes times out, and is already
checked upstream.
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Copy link
Copy Markdown
Collaborator

@drbh drbh left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

really cool! lgtm

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.

Add JAX glue for tvm-ffi

3 participants