We don’t have any docs about custom init functions, if you have any ideas for additions there they would be very welcome. For now I would recommend following the pattern used by functions in Utility Functions · Flux. TL;DR all you need is a function that takes dims and returns an initialized array.