Tag: JAX

A First Look at JAX

A First Look at JAX

A First Look at JAX A First Look at JAX Put on your metaphorical safety goggles and start building something weird with JAX. Source: www.pragmatic.ml/first-look-at-jax/ …

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 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 …

Subscribe to our Digest