N8python / mlx-pretrainLinks
A simple MLX implementation for pretraining LLMs on Apple Silicon.
☆85Updated 5 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)☆61Updated last year
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated last year
- ☆67Updated 8 months ago
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆118Updated last year
- look how they massacred my boy☆63Updated last year
- smolLM with Entropix sampler on pytorch☆149Updated last year
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆110Updated 11 months ago
- MLX Transformers is a library that provides model implementation in MLX. It uses a similar model interface as HuggingFace Transformers an…☆72Updated last year
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆59Updated 3 months ago
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆77Updated 11 months ago
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆37Updated last year
- NanoGPT-speedrunning for the poor T4 enjoyers☆73Updated 9 months ago
- An easy-to-understand framework for LLM samplers that rewind and revise generated tokens☆150Updated 3 weeks ago
- Fast parallel LLM inference for MLX☆245Updated last year
- A collection of optimizers for MLX☆54Updated last month
- Official CLI and Python SDK for Prime Intellect - access GPU compute, remote sandboxes, RL environments, and distributed training infrast…☆148Updated this week
- Plotting (entropy, varentropy) for small LMs☆99Updated 8 months ago
- Train your own SOTA deductive reasoning model☆107Updated 11 months ago
- Benchmarks comparing PyTorch and MLX on Apple Silicon GPUs☆93Updated 2 weeks ago
- An AI character interaction system with emotional modeling and advanced memory management☆17Updated last year
- run embeddings in MLX☆97Updated last year
- An introduction to LLM Sampling☆79Updated last year
- EXO Gym is an open-source Python toolkit that facilitates distributed AI research.☆94Updated 2 months ago
- Project code for training LLMs to write better unit tests + code☆21Updated 8 months ago
- Train Large Language Models on MLX.☆245Updated last week
- 🤖 Complete reproduction of 'AlphaGo Moment for Model Architecture Discovery' using MLX-LM instead of GPT-4. Autonomous neural architectu…☆27Updated 6 months ago
- Lego for GRPO☆30Updated 8 months ago
- Very basic framework for composable parameterized large language model (Q)LoRA / (Q)Dora fine-tuning using mlx, mlx_lm, and OgbujiPT.☆43Updated 7 months ago
- smolbox of recipies☆29Updated 9 months ago
- ☆137Updated last year