N8python / mlx-pretrainLinks
A simple MLX implementation for pretraining LLMs on Apple Silicon.
☆84Updated 2 months ago
Alternatives and similar repositories for mlx-pretrain
Users that are interested in mlx-pretrain are comparing it to the libraries listed below
Sorting:
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆62Updated last year
- ☆68Updated 5 months ago
- look how they massacred my boy☆63Updated last year
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆114Updated last year
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆107Updated 8 months ago
- Plotting (entropy, varentropy) for small LMs☆98Updated 5 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated last year
- Fast parallel LLM inference for MLX☆227Updated last year
- Train Large Language Models on MLX.☆213Updated last week
- Train your own SOTA deductive reasoning model☆108Updated 8 months ago
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆72Updated 9 months ago
- smolLM with Entropix sampler on pytorch☆150Updated last year
- run embeddings in MLX☆96Updated last year
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆58Updated 3 weeks ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Updated 6 months ago
- Explore a simple example of utilizing MLX for RAG application running locally on your Apple Silicon device.☆177Updated last year
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆33Updated last year
- smol models are fun too☆93Updated last year
- Distributed Inference for mlx LLm☆98Updated last year
- Implementation of nougat that focuses on processing pdf locally.☆83Updated 9 months ago
- ☆116Updated 10 months ago
- Benchmarks comparing PyTorch and MLX on Apple Silicon GPUs☆89Updated last year
- MLX Transformers is a library that provides model implementation in MLX. It uses a similar model interface as HuggingFace Transformers an…☆67Updated 11 months ago
- ☆67Updated last year
- An easy-to-understand framework for LLM samplers that rewind and revise generated tokens☆145Updated 8 months ago
- An AI character interaction system with emotional modeling and advanced memory management☆17Updated last year
- A collection of lightweight interpretability scripts to understand how LLMs think☆63Updated this week
- Compiling useful links, papers, benchmarks, ideas, etc.☆45Updated 7 months ago
- An introduction to LLM Sampling☆79Updated 10 months ago
- explore token trajectory trees on instruct and base models☆148Updated 5 months ago