JAX-LM Assignment
The assignment companion for JAX-LM
Chuyi Shang
A personal website for Chuyi Shang.
Latest posts
JAX-LM: Language Modelling and Distributed Training in JAX
A guide to language modelling and distributed training from scratch in JAX
A personal website for Chuyi Shang.
The assignment companion for JAX-LM
A guide to language modelling and distributed training from scratch in JAX