stockeh / mlx-optimizers
A collection of optimizers for MLX
☆34Updated 3 weeks ago
Alternatives and similar repositories for mlx-optimizers:
Users that are interested in mlx-optimizers are comparing it to the libraries listed below
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆91Updated last month
- MLX Transformers is a library that provides model implementation in MLX. It uses a similar model interface as HuggingFace Transformers an…☆65Updated 4 months ago
- Very basic framework for composable parameterized large language model (Q)LoRA / (Q)Dora fine-tuning using mlx, mlx_lm, and OgbujiPT.☆37Updated last month
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆39Updated 2 months ago
- run embeddings in MLX☆85Updated 6 months ago
- mlx image models for Apple Silicon machines☆76Updated 4 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆28Updated 2 months ago
- MLX Swift implementation of Andrej Karpathy's Let's build GPT video☆57Updated 11 months ago
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆25Updated 9 months ago
- [WIP] Transformer to embed Danbooru labelsets☆13Updated last year
- Latent Large Language Models☆17Updated 7 months ago
- Lego for GRPO☆26Updated this week
- BH hackathon☆14Updated last year
- Find out why your CoreML model isn't running on the Neural Engine!☆25Updated 9 months ago
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆68Updated last month
- Using multiple LLMs for ensemble Forecasting☆16Updated last year
- look how they massacred my boy☆63Updated 5 months ago
- KAN (Kolmogorov–Arnold Networks) in the MLX framework for Apple Silicon☆16Updated 3 weeks ago
- Simple GRPO scripts and configurations.☆58Updated 2 months ago
- ☆27Updated 8 months ago
- ☆60Updated last year
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆63Updated 5 months ago
- Collection of autoregressive model implementation☆83Updated last month
- Scripts to create your own moe models using mlx☆89Updated last year
- ☆38Updated 8 months ago
- Ongoing research training transformer models at scale☆36Updated last year
- A python command-line tool to download & manage MLX AI models from Hugging Face.☆17Updated 7 months ago
- alternative way to calculating self attention☆18Updated 10 months ago
- ☆66Updated 10 months ago
- ☆48Updated last year