Found 25 repositories(showing 25)
kingoflolz
Model parallel transformers in JAX and Haiku
Lisennlp
mesh_transformer_jax + easylm train llama model
mosmos6
test for v2
mosmos6
v2
mosmos6
jax 0.4.26
mosmos6
v5p
mosmos6
finetune test of jax4 repo
mosmos6
jax4 core plus original
mosmos6
test for v3 jax0.3.5 plain text
mosmos6
shmap&v5e-1
mosmos6
test
siewkagaming
No description available
mosmos6
test for v3
mosmos6
shmap
mosmos6
test for v3 jax0.3.25
mosmos6
xmap&v5e-1
timwillhack
No description available
Deep-Fold
No description available
okbuddyhololive
A fork of a library in JAX for model parallelism of transformers, for "Project Cybertard"
siewkagaming
No description available
alexyaluninsber
No description available
iamianM
No description available
Yakultie
A fork of kingoflolz/mesh-transformer-jax to train GPT with CLIP supervision
EleutherAI
Experiments pertaining to visually grounding Neo, built off of mesh-transformer-jax
mosmos6
This is a modified version of https://github.com/VE-FORBRYDERNE/mesh-transformer-jax/tree/ck to be adapted to JAX 0.3.25 so this runs on colab with TPU_driver0.2
All 25 repositories loaded