Skip to content

[DRAFT] Add framework-agnostic operator APIs with JAX support#363

Draft
mgoldfarb-nvidia wants to merge 13 commits into
NVIDIA:developfrom
mgoldfarb-nvidia:mgoldfarb/jax-vanilla-gemms
Draft

[DRAFT] Add framework-agnostic operator APIs with JAX support#363
mgoldfarb-nvidia wants to merge 13 commits into
NVIDIA:developfrom
mgoldfarb-nvidia:mgoldfarb/jax-vanilla-gemms

Refactor JAX kernel calls around tensor descriptors

8386cb6
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs