The authors of Critical initialization of wide and deep neural networks through partial Jacobians: general theory and applications to LayerNorm have not publicly listed the code yet.

Request code directly from the authors: Ask Authors for Code

Get an expert to implement this paper: Request Implementation

(OR if you have code to share with the community, please add it here 😊🙏 )