submartingales's starred repositories
mesh-transformer-jax
Model parallel transformers in JAX and Haiku
annotation-abuse
Rust-like macros in Python by abusing type annotations
Model parallel transformers in JAX and Haiku
Rust-like macros in Python by abusing type annotations