Reactant + Lux LLama on TPUs

yolhan_mannes
Sat, 09 Aug 2025 15:03:53 +0000
General Usage

Just wanted to share how to use Reactant+Lux on TPUs for a LLama like transformer model, it seems really nice to write even though its hard to adapt to the compact way of writting it does feel a lot better after. Still cannot take gradients though but this may be known.

colab.research.google.com

Google Colab

Didn’t try to mesh the devices yet but it may be even better after

1 post - 1 participant