flax
Fix colab & github links
#2363
Merged

Loading