N8python / mlx-pretrainLinks
A simple MLX implementation for pretraining LLMs on Apple Silicon.
☆81Updated 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)☆64Updated 8 months ago
- look how they massacred my boy☆63Updated 8 months ago
- ☆64Updated last month
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated 8 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆101Updated 4 months ago
- run embeddings in MLX☆90Updated 9 months ago
- rl from zero pretrain, can it be done? we'll see.☆65Updated 2 weeks ago
- smolLM with Entropix sampler on pytorch☆150Updated 8 months ago
- Plotting (entropy, varentropy) for small LMs☆97Updated last month
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆70Updated 5 months ago
- Train Large Language Models on MLX.☆126Updated this week
- Fast parallel LLM inference for MLX☆198Updated last year
- Explore a simple example of utilizing MLX for RAG application running locally on your Apple Silicon device.☆172Updated last year
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆110Updated last year
- ☆75Updated 6 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆54Updated 5 months ago
- Distributed Inference for mlx LLm☆93Updated 11 months ago
- ☆115Updated 6 months ago
- Lego for GRPO☆28Updated last month
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆32Updated last year
- MLX-Embeddings is the best package for running Vision and Language Embedding models locally on your Mac using MLX.☆177Updated last month
- Benchmarks comparing PyTorch and MLX on Apple Silicon GPUs☆87Updated 11 months ago
- ☆66Updated last year
- MLX Transformers is a library that provides model implementation in MLX. It uses a similar model interface as HuggingFace Transformers an…☆66Updated 7 months ago
- explore token trajectory trees on instruct and base models☆134Updated last month
- Train your own SOTA deductive reasoning model☆96Updated 4 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆68Updated 2 months ago
- A pure MLX-based training pipeline for fine-tuning LLMs using GRPO on Apple Silicon.☆42Updated 5 months ago
- Very basic framework for composable parameterized large language model (Q)LoRA / (Q)Dora fine-tuning using mlx, mlx_lm, and OgbujiPT.☆42Updated 3 weeks ago
- Minimal, clean code implementation of RAG with mlx using gguf model weights☆52Updated last year