Submitted by jakiwjakiw t3_xwam5y in MachineLearning
jakiwjakiw OP t1_ir6lny2 wrote
Reply to comment by Small_Stand_8716 in [R] Introduction to Diffusion Models in JAX by jakiwjakiw
I mainly used flax since most of the code I was reading myself was also using flax, but I wouldn't be able to objectively compare them since I haven't used haiku yet. So can't help, sorry!
Small_Stand_8716 t1_iraj2sy wrote
Thank you for your reply!
Viewing a single comment thread. View all comments