git404hub

what is nemotron-puzzle-mlx fr?

sxuff/nemotron-puzzle-mlx — explained in plain English

Analysis updated 2026-05-18

0PythonAudience · researcherComplexity · 5/5LicenseSetup · hard

tl;dr

Tools and a guide for running a large Nemotron AI model on a 64GB Apple Silicon Mac using compressed, mixed precision weights.

vibe map

mindmap
  root((Nemotron Puzzle MLX))
    What it does
      Runs 75B model locally
      Mixed precision quantization
      Benchmarks results
    Tech stack
      Python
      MLX
      Apple Silicon
    Use cases
      Local LLM inference
      Quantization comparison
      Checkpoint conversion
    Audience
      ML researchers
      Apple Silicon users

Code map

Detail Auto

An interactive map of this repo's files and how they connect — its source is parsed live in your browser. Click Visualize to build it.

filefunction / class

what do people make with this?

VIBE 1

Run a 75B parameter language model locally on a Mac instead of the cloud.

VIBE 2

Compare different quantization bit widths for memory, speed, and accuracy tradeoffs.

VIBE 3

Convert an original model checkpoint into a compressed MLX format yourself.

VIBE 4

Benchmark a local model against a fixed set of reasoning and formatting tasks.

what's the stack?

PythonMLXApple Silicon

how it stacks up fr

sxuff/nemotron-puzzle-mlx0xallam/my-recipe0xhassaan/nn-from-scratch
Stars00
LanguagePythonPythonPython
Last pushed2022-11-22
MaintenanceDormant
Setup difficultyhardmoderatemoderate
Complexity5/52/54/5
Audienceresearchergeneraldeveloper

Figures from each repo's GitHub metadata at analysis time.

how do i run it?

Difficulty · hard time til it works · 1h+

Requires a 64GB Apple Silicon Mac, a custom MLX branch, and downloading a large model checkpoint.

The repository's own code is MIT licensed, but the model weights it works with have a separate license.

in plain english

This project lets you run a large language model called Nemotron Puzzle 75B on an Apple Silicon Mac with 64 GB of memory, using a technique called mixed precision quantization to shrink the model enough to fit. Quantization means storing the model's numbers with fewer bits than normal, which makes the file smaller and faster to run, at some cost to accuracy. The model is a mixture of experts design, meaning most of its size comes from a large bank of specialized sub networks that are only partly used for any given word it generates. The author found that applying a more aggressive four bit compression specifically to that expert bank, while keeping the rest of the model at a higher precision, gave the best balance. Tested on an Apple M2 Max, this four bit expert version produced a smaller file, faster generation speed, lower memory use, and better results on a set of test tasks compared to a five bit version. To use it, you need support for this specific model type added to the underlying MLX library, which the author has proposed as a pull request and made available as a separate branch you can install in the meantime. You can either download the already converted model checkpoint from Hugging Face, or convert your own copy of the original model yourself using the included scripts. The repository also includes scripts to verify that generation is working correctly and to run a set of benchmark checks comparing different quantization settings. A notable part of this project is a bug fix the author found and documented, where a specific calculation inside the model needed to be done in a higher precision format to match the original results correctly, which they explain in detail in an included engineering report. The repository itself contains only code and benchmark results, not the actual model weights, which must be obtained separately and are subject to their own license. The code in this repository is released under the MIT License.

prompts (copy fr)

prompt 1
Explain why mixed precision quantization helps a mixture of experts model.
prompt 2
Help me install the custom MLX branch this project depends on.
prompt 3
Walk me through converting my own copy of the Nemotron checkpoint.
prompt 4
Show me how the Mamba timestep precision bug was fixed in this project.

Frequently asked questions

what is nemotron-puzzle-mlx fr?

Tools and a guide for running a large Nemotron AI model on a 64GB Apple Silicon Mac using compressed, mixed precision weights.

What language is nemotron-puzzle-mlx written in?

Mainly Python. The stack also includes Python, MLX, Apple Silicon.

What license does nemotron-puzzle-mlx use?

The repository's own code is MIT licensed, but the model weights it works with have a separate license.

How hard is nemotron-puzzle-mlx to set up?

Setup difficulty is rated hard, with roughly 1h+ to a first successful run.

Who is nemotron-puzzle-mlx for?

Mainly researcher.

peek the repo → explain another one

This repo across BitVibe Labs

double-check against the repo, no cap.