# salesforce/jaxformer

<https://repo.masters-masters.com/en/repos/salesforce/jaxformer>

[Home](https://repo.masters-masters.com/en) › [salesforce](https://repo.masters-masters.com/en/owners/salesforce) › [jaxformer](https://repo.masters-masters.com/en/repos/salesforce/jaxformer)

salesforce/jaxformer has 300 stars on GitHub. Its RepoMasters card: Alpha tier. Minimal library to train LLMs on TPU in JAX with pjit(). Language: Python.

| Field | Value |
| --- | --- |
| Owner | [salesforce](https://repo.masters-masters.com/en/owners/salesforce) |
| Stars | 300 |
| Forks | 39 |
| Language | [Python](https://repo.masters-masters.com/en/languages/python) |
| License | BSD-3-Clause |
| Created | 2022 |
| Tier | [Alpha (250 to 699 ★)](https://repo.masters-masters.com/en/ranks/b_movie) |
| View on GitHub | [salesforce/jaxformer](https://github.com/salesforce/jaxformer) |

## Repository

Minimal library to train LLMs on TPU in JAX with pjit().

The tier for repos with 250 to 699 ★. Published odds per card pulled: 20%. Rarity: Uncommon.

## More repos by salesforce

- [salesforce/LAVIS · ★ 11,256 · Jupyter Notebook · Stable](https://repo.masters-masters.com/en/repos/salesforce/LAVIS)
- [salesforce/BLIP · ★ 5,703 · Jupyter Notebook · Stable](https://repo.masters-masters.com/en/repos/salesforce/BLIP)
- [salesforce/CodeGen · ★ 5,181 · Python · Stable](https://repo.masters-masters.com/en/repos/salesforce/CodeGen)
- [salesforce/Merlion · ★ 4,488 · Python · RC](https://repo.masters-masters.com/en/repos/salesforce/Merlion)
- [salesforce/akita · ★ 3,661 · TypeScript · RC](https://repo.masters-masters.com/en/repos/salesforce/akita)
- [salesforce/ja3 · ★ 3,097 · Python · RC](https://repo.masters-masters.com/en/repos/salesforce/ja3)

## Nearby in Python

- [dankumarlab2090u1/minecraft-hack-menu-2026 · ★ 300 · Python · Alpha](https://repo.masters-masters.com/en/repos/dankumarlab2090u1/minecraft-hack-menu-2026)
- [davidlopez-x05x8/minecraft-aim-assist · ★ 300 · Python · Alpha](https://repo.masters-masters.com/en/repos/davidlopez-x05x8/minecraft-aim-assist)
- [SplendorHornbill/windows-animated-wallpapers · ★ 300 · Python · Alpha](https://repo.masters-masters.com/en/repos/SplendorHornbill/windows-animated-wallpapers)
- [iamteem/redisco · ★ 300 · Python · Alpha](https://repo.masters-masters.com/en/repos/iamteem/redisco)
- [zacharyvoase/jsonpipe · ★ 300 · Python · Alpha](https://repo.masters-masters.com/en/repos/zacharyvoase/jsonpipe)
- [allfro/pymetasploit · ★ 300 · Python · Alpha](https://repo.masters-masters.com/en/repos/allfro/pymetasploit)

---
Data: GitHub · Illustration drawn by RepoMasters from the repo’s ID and its language color.