Neural Networks in pure JAX (with automatic differentiation)

[@DefinitelyNotAMachineCultist]

Thanks a bunch. I can never find enough JAX tuts lol Even large language models like ChatGPT have outdated info a lot of the time due to their current knowledge cutoff dates. It's just this and the docs for me hehe...

[@MachineLearningSimulation]

Glad I could help! :)

[@juancolmenares6185]

Very nice explanation!

[@MachineLearningSimulation]

Thanks a lot 😊

[@christiansinger2497]

Very informative video as always!

[@MachineLearningSimulation]

Thanks a lot 😊

[@minhphan8552]

Thanks a lot for the nice video! I have a naive question. The layers of the neural network seems similar to a polynomial. How is a neural network better than a polynomial fit?

Great, seems like magic. Thanks!

[@MachineLearningSimulation]

Nice, thanks :). Indeed, JAX is really smooth to use. I find the AD interface better than in tensorflow or Pytorch.