jax
[WIP] A minimal public API for custom primitives
#24726

Open

[WIP] A minimal public API for custom primitives #24726

dfm wants to merge 1 commit into jax-ml:main from dfm:custom-primitive

dfm assigned

dfm 1 year ago

dfm force pushed from 7d8cea35 to a6c33cc9 1 year ago

A public API for custom primitives.

e19b6d39

dfm force pushed from a6c33cc9 to e19b6d39 1 year ago

Reviewers

No reviews

Assignees

dfm

Labels

None yet

Milestone

No milestone