Samuel Ainsworth
@skainswo.bsky.social
prev: @BrownUniversity, @uwcse/@uw_wail phd, ex-@cruise, RS @waymo. 0.1x engineer, 10x friend.
spondyloarthritis, cars ruin cities, open source
spondyloarthritis, cars ruin cities, open source
Adding support for random ops and batch norm required a near complete rewrite of the library internals. v0.1.0 now makes extensive use of PyTorch Modes (pytorch.org/docs/stable/...), an underrating part of the PyTorch API IMHO.
Shout out to Nick Boyd who inspired me to undertake this rewrite!
Shout out to Nick Boyd who inspired me to undertake this rewrite!
Extending PyTorch — PyTorch 2.6 documentation
pytorch.org
February 27, 2025 at 2:17 AM
Adding support for random ops and batch norm required a near complete rewrite of the library internals. v0.1.0 now makes extensive use of PyTorch Modes (pytorch.org/docs/stable/...), an underrating part of the PyTorch API IMHO.
Shout out to Nick Boyd who inspired me to undertake this rewrite!
Shout out to Nick Boyd who inspired me to undertake this rewrite!
As a reminder, torch2jax enables running PyTorch code in JAX. Mix-and-match PyTorch and JAX code with seamless, end-to-end autodiff, use JAX classics like jit, grad, and vmap on PyTorch code, and run PyTorch models on TPUs.
See x.com/SamuelAinswo... for the initial project announcement.
See x.com/SamuelAinswo... for the initial project announcement.
February 27, 2025 at 2:17 AM
As a reminder, torch2jax enables running PyTorch code in JAX. Mix-and-match PyTorch and JAX code with seamless, end-to-end autodiff, use JAX classics like jit, grad, and vmap on PyTorch code, and run PyTorch models on TPUs.
See x.com/SamuelAinswo... for the initial project announcement.
See x.com/SamuelAinswo... for the initial project announcement.