Hacker News with Generative AI: JAX

SDFs and the Fast sweeping algorithm in Jax (rohangautam.github.io)
This is going to be a fun blog - we'll explore the intuition behind level sets, the Eikonal equation, and implement a speedy algorithm for solving this equation, called the fast sweeping method, in JAX.
Mathematics, Computer Science, Algorithms, JAX
39 points by beansbeansbeans 15 days ago | 7 comments
Understanding Automatic Differentiation in Jax: A Deep Dive (ispeakcode.substack.com)
Welcome to the world of JAX, where differentiation happens automatically, faster than a caffeine-fuelled coder at 3 a.m.! In this post, we’re going to delve into the concept of Automatic Differentiation (AD), a feature at the heart of JAX, and we’ll explore why it’s such a game changer for machine learning, scientific computing, and any other context where derivatives matter. The popularity of JAX has been increasing lately, thanks to the emerging field of scientific machine learning powered by differentiable programming.
Automatic Differentiation, Machine Learning, Scientific Computing, Programming, JAX
3 points by lawrenceyan 93 days ago | 0 comments