Llama and Falcon in JAX #17667
Replies: 5 comments 2 replies
-
Here's my haiku port of Llama: https://github.com/davisyoshida/llama-haiku Haiku has perished though 💔 |
Beta Was this translation helpful? Give feedback.
-
Looks like you can find this in Levanter/Equinox: stanford-crfm/levanter#270 |
Beta Was this translation helpful? Give feedback.
-
Greetings from the author of llama-2-jax 👋🏻 |
Beta Was this translation helpful? Give feedback.
-
Sorry, I forgot to include this one - https://github.com/openlm-research/open_llama |
Beta Was this translation helpful? Give feedback.
-
A bit late to the party, but there is falcon in jax now https://github.com/defdet/falcon-jax |
Beta Was this translation helpful? Give feedback.
-
So far I've found at least 4 ports of Llama to JAX:
I'm wondering, are there are others that folks have done?
Also, I've looked for ports of Falcon to JAX and so far I haven't found any. Do you know of any?
Beta Was this translation helpful? Give feedback.
All reactions