Jax Ml Frameworks Transformer Models
There are 3 jax ml frameworks models tracked. The highest-rated is Sea-Snell/JAXSeq at 39/100 with 210 stars.
Get all 3 projects as JSON
curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=transformers&subcategory=jax-ml-frameworks&limit=20"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
| # | Model | Score | Tier |
|---|---|---|---|
| 1 |
Sea-Snell/JAXSeq
Train very large language models in Jax. |
|
Emerging |
| 2 |
haoliuhl/instructrl
Instruction Following Agents with Multimodal Transforemrs |
|
Emerging |
| 3 |
jrajath94/jax-transformer-impl
JAX/XLA Transformer with MHA, MQA, GQA (Ainslie et al. 2023) — JIT, vmap, pmap |
|
Experimental |