Samarendra Dash

Results 4 comments of Samarendra Dash

If this issue is open to contributions, I would like to pick this up.

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...