From PyTorch to JAX: towards neural net frameworks that purify stateful code — Sabrina J. Mielke

From PyTorch to JAX: towards neural net frameworks that purify stateful code — Sabrina J. Mielke

From PyTorch to JAX: towards neural net frameworks that purify stateful code — Sabrina J. Mielke

Hi. I’m a PhD student at Johns Hopkins University Center for Language and Speech Processing (JHU CLSP). Machine learning and natural language are fascinating.

Source: sjmielke.com/jax-purify.htm