flax
ab2efb97 - Added preferred_element_type arg to nnx.Linear*, nnx.Conv*, nnx.Einsum

Commit
123 days ago
Added preferred_element_type arg to nnx.Linear*, nnx.Conv*, nnx.Einsum Fixes #4890
Author
Committer
Parents
Loading