Samarendra Dash
Results
4
comments of
Samarendra Dash
If this issue is open to contributions, I would like to pick this up.
I can take up the issue if approved.
Would it be possible to add a feature that wraps JAX initializers into Penzai initializers? Currently, Penzai only has two implemented initializers for neural networks. I believe the implementation would...
I am working on it! :)