Hello Juan,
I found your post "Flax and Numpyro Toy Example" very cool and interesting. As a PYMC user I wanted to see if I could achieve similar results using PYMC. You can see my implementation at the very end of this notebook:
https://gist.github.com/Dekermanjian/29c785306c1ecccd2f1572aac8eecc75
It would be wonderful if you would be open to add this in some way to your blog to give the PYMC users a way to incorporate neural networks in their projects. If you are open please feel free to modify anything as you please.
Hello Juan,
I found your post "Flax and Numpyro Toy Example" very cool and interesting. As a PYMC user I wanted to see if I could achieve similar results using PYMC. You can see my implementation at the very end of this notebook:
https://gist.github.com/Dekermanjian/29c785306c1ecccd2f1572aac8eecc75
It would be wonderful if you would be open to add this in some way to your blog to give the PYMC users a way to incorporate neural networks in their projects. If you are open please feel free to modify anything as you please.