Pomni
Current Frontier LLMs are really bad at assisting with writing JAX code due to the sparse resources spread across multiple different platforms.
Pomni is a finetune of a gemma model on JAX data, meant to run locally. It provides a TUI interface making it easy to integrate into existing terminal workflows.
Setup
To use the finetuned pomni model, you can simply do:
pip install pomni
pomni
to automatically download and run the model.
Screenshot
Metadata
Release files for pomni 0.1.7
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| pomni-0.1.7.tar.gz | 177.4 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| pomni-0.1.7-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 355.0 kB
Release files / pomni-0.1.7.tar.gz
| Download URL | pomni-0.1.7.tar.gz |
|---|---|
| Size | 177.4 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
1fbd0a040e0d2df9afee70fb99af1ff012ead73363fc3118005ff45309e352db
|
|
BLAKE2b-256 checksum How to use checksums |
24562a13001b84a1872554beb89ec7f9ceb7a9316e68538cbbd2a3a44925e255
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
uv/0.7.3
|
Release files / pomni-0.1.7-py3-none-any.whl
| Download URL | pomni-0.1.7-py3-none-any.whl |
|---|---|
| Size | 177.6 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
b2924b13a8645b8d696adb5af1f0f0a9820fd80cf7e558f74391068159b32f69
|
|
BLAKE2b-256 checksum How to use checksums |
22b2a644f975e773043c5611c007eefd1dd072c14bf5831e19ea09aaa5e5e09f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
uv/0.7.3
|