yitongx

Results 2 issues of 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...

documentation

Hi, first of all thanks for sharing your improved codes! While I was testing your jupyter notebook on CoLab, I have the following problems: Firstly, I would like to know...