N8python / mlx-pretrainLinks
A simple MLX implementation for pretraining LLMs on Apple Silicon.
☆76Updated last month
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 6 months ago
- look how they massacred my boy☆63Updated 7 months ago
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆68Updated 3 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆66Updated last month
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆100Updated 2 months ago
- Train Large Language Models on MLX.☆77Updated this week
- ☆57Updated last week
- Lego for GRPO☆28Updated this week
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆50Updated 6 months ago
- An introduction to LLM Sampling☆78Updated 5 months ago
- A collection of optimizers for MLX☆36Updated this week
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆53Updated 3 months ago
- A graph visualization of attention☆55Updated last week
- smolLM with Entropix sampler on pytorch☆150Updated 7 months ago
- Train your own SOTA deductive reasoning model☆92Updated 2 months ago
- Fast parallel LLM inference for MLX☆189Updated 10 months ago
- A pure MLX-based training pipeline for fine-tuning LLMs using GRPO on Apple Silicon.☆39Updated 4 months ago
- run embeddings in MLX☆89Updated 8 months ago
- MLX Transformers is a library that provides model implementation in MLX. It uses a similar model interface as HuggingFace Transformers an…☆65Updated 6 months ago
- MLX Swift implementation of Andrej Karpathy's Let's build GPT video☆57Updated last year
- ☆113Updated 5 months ago
- Entropy Based Sampling and Parallel CoT Decoding☆17Updated 7 months ago
- ☆93Updated 7 months ago
- Plotting (entropy, varentropy) for small LMs☆96Updated last week
- Training an LLM to use a calculator with multi-turn reinforcement learning, achieving a **62% absolute increase in evaluation accuracy**.☆38Updated 3 weeks ago
- Distributed Inference for mlx LLm☆92Updated 10 months ago
- Synthetic data derived by templating, few shot prompting, transformations on public domain corpora, and monte carlo tree search.☆32Updated 3 months ago
- smol models are fun too☆92Updated 6 months ago
- KAN (Kolmogorov–Arnold Networks) in the MLX framework for Apple Silicon☆18Updated last month
- MLX implementation of xLSTM model by Beck et al. (2024)☆27Updated 11 months ago