ElleLeonne / handmade_flax_transformers Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

A basic transformer architecture in flax. XLA compiler go brrrrr

0 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
README.md		README.md
axial_vit.py		axial_vit.py
llama_model.py		llama_model.py

Repository files navigation

flax_transformers

Some basic transformer architectures in flax. XLA compiler go brrrrr

Leftovers from my tinkering. May need some debugging still.

About

A basic transformer architecture in flax. XLA compiler go brrrrr

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%