zach

Results 2 comments of zach

I replaced line 18 with `from flax import linen as nn` and then the notebook ran as expected I just submitted a pull request to fix this

Apologies for not being more specific, it constantly prints that and is unable to detect the fans.