jax
81ebe31f - [hijax] test vmap + hijax types + hijax primitive

Commit

100 days ago

[hijax] test vmap + hijax types + hijax primitive Co-authored-by: Yash Katariya <yashkatariya@google.com>

References

Author

mattjj

Committer

mattjj

Parents