yitongx

Hi jax team. Thank you for your great framework. I noticed that there is neither explaination on how to use jax.tree_util.tree_reduce() nor on its argument type. Would you please explain...