jax-ml / jax-llm-examples
Minimal yet performant LLM examples in pure JAX
Why this job: matched by what the repo talks about: llm, pure, yet
What it does
A collection (in progress) of example high-performance large language model implementations, written with JAX.
Get started
No single install command. Read the steps in the README (opens in new tab)
- Maintenance
- Active
- Last commit 2 months ago
- Stars
- 287
- +1 this week
- Licence
- Apache-2.0
- Allows business use
- Downloads for
- None listed
- No release files
- Latest version
- None yet
- No releases
- Built with
- Python
- 99% of the code
Is it looked after?
- Contributors
- 1
- Last 90 days
- Issues
- 18% closed
- 9 still open
- Pull requests
- 32 merged
- 0 waiting
- Releases
- None
- Never published
- Contributing guide (opens in new tab)
- No security policy
- No code of conduct
Similar repos
| Repo | Stars | Last commit | Licence | Downloads for |
|---|---|---|---|---|
| foldl / chatllm.cpp | 942 | 16 days ago | MIT | No downloads |
| liguodongiot / llm-action | 25,274 | 29 days ago | Apache-2.0 | No downloads |
| AI-Hypercomputer / maxtext | 2,460 | today | Apache-2.0 | No downloads |
More: languages, topics and project record
Languages
- Python 99%
- Shell 1%
Topics
- jax
- llm
- llm-inference
Project record
- JobLocal LLMs
- StatusListed
- Commits116
- Forks39
- Watching6
- Stars in 28 days+2
- CreatedFeb 6, 2025
- First seen by usNov 7, 2026
- CheckedNov 7, 2026