I am a research scientist working in AI x Biology at Google X

My Latest Articles

Why jax.lax.scan Is Slow on GPU and How Pallas Fixes It

A Simple Walk Through of Equinox for Building Models in JAX

Diffrax Secrets or How to Build a Differentiable ODE solver from Scratch in JAX

Contact

<aside> <img src="/icons/sharing_purple.svg" alt="/icons/sharing_purple.svg" width="40px" /> LinkedIn

</aside>

<aside> <img src="/icons/git_purple.svg" alt="/icons/git_purple.svg" width="40px" /> GitHub

</aside>

<aside> 📩 wanglin44 at gmail

</aside>