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

References

Author

vfdev-5

Committer

vfdev-5

Parents

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