N8python / mlx-pretrainLinks
A simple MLX implementation for pretraining LLMs on Apple Silicon.
☆83Updated 3 weeks 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 10 months ago
- look how they massacred my boy☆64Updated 10 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆105Updated 6 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated 10 months ago
- Plotting (entropy, varentropy) for small LMs☆98Updated 3 months ago
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆112Updated last year
- ☆68Updated 3 months ago
- smolLM with Entropix sampler on pytorch☆150Updated 10 months ago
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆73Updated 7 months ago
- Fast parallel LLM inference for MLX☆216Updated last year
- 🤖 Complete reproduction of 'AlphaGo Moment for Model Architecture Discovery' using MLX-LM instead of GPT-4. Autonomous neural architectu…☆23Updated last month
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆55Updated 7 months ago
- Train your own SOTA deductive reasoning model☆106Updated 6 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆71Updated 4 months ago
- run embeddings in MLX☆91Updated 11 months ago
- An easy-to-understand framework for LLM samplers that rewind and revise generated tokens☆146Updated 6 months ago
- MLX Transformers is a library that provides model implementation in MLX. It uses a similar model interface as HuggingFace Transformers an…☆66Updated 9 months ago
- Explore a simple example of utilizing MLX for RAG application running locally on your Apple Silicon device.☆175Updated last year
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆32Updated last year
- A collection of optimizers for MLX☆52Updated last week
- Project code for training LLMs to write better unit tests + code☆21Updated 3 months ago
- Train Large Language Models on MLX.☆159Updated last month
- Benchmarks comparing PyTorch and MLX on Apple Silicon GPUs☆88Updated last year
- explore token trajectory trees on instruct and base models☆132Updated 3 months ago
- Simple Transformer in Jax☆139Updated last year
- MLX-Embeddings is the best package for running Vision and Language Embedding models locally on your Mac using MLX.☆195Updated last week
- smol models are fun too☆93Updated 10 months ago
- ☆116Updated 8 months ago
- Entropy Based Sampling and Parallel CoT Decoding☆17Updated 11 months ago
- MLX implementation of xLSTM model by Beck et al. (2024)☆28Updated last year