TristanBilot / mlx-GCN
MLX implementation of GCN, with benchmark on MPS, CUDA and CPU (M1 Pro, M2 Ultra, M3 Max).
☆24Updated last year
Alternatives and similar repositories for mlx-GCN
Users that are interested in mlx-GCN are comparing it to the libraries listed below
Sorting:
- Clean RL implementation using MLX☆30Updated last year
- Package of useful sampling algorithms written in MLX.☆17Updated last year
- A collection of optimizers for MLX☆35Updated this week
- KAN (Kolmogorov–Arnold Networks) in the MLX framework for Apple Silicon☆16Updated last week
- A fast minimalistic implementation of guided generation on Apple Silicon using Outlines and MLX☆53Updated last year
- Very basic framework for composable parameterized large language model (Q)LoRA / (Q)Dora fine-tuning using mlx, mlx_lm, and OgbujiPT.☆40Updated 2 months ago
- MLX implementation of xLSTM model by Beck et al. (2024)☆27Updated 11 months ago
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆26Updated 10 months ago
- Minimal, clean code implementation of RAG with mlx using gguf model weights☆50Updated last year
- mlx image models for Apple Silicon machines☆78Updated last month
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated last month
- Graph Neural Network library made for Apple Silicon☆189Updated 7 months ago
- A python command-line tool to download & manage MLX AI models from Hugging Face.☆17Updated 8 months ago
- Implementation of Spectral State Space Models☆16Updated last year
- Roberta Question Answering using MLX.☆24Updated last year
- BH hackathon☆14Updated last year
- Training hybrid models for dummies.☆21Updated 3 months ago
- Collection of LLM completions for reasoning-gym task datasets☆20Updated this week
- The official evaluation suite and dynamic data release for MixEval.☆11Updated 7 months ago
- ☆27Updated 10 months ago
- Benchmarks comparing PyTorch and MLX on Apple Silicon GPUs☆79Updated 9 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆74Updated last week
- Shared personal notes created while working with the Apple MLX machine learning framework☆23Updated 10 months ago
- aesthetic tensor visualiser☆16Updated 2 weeks ago
- Portfolio REgret for Confidence SEquences☆14Updated 5 months ago
- ☆14Updated 11 months ago
- OmegaViT (ΩViT) is a cutting-edge vision transformer architecture that combines multi-query attention, rotary embeddings, state space mod…☆14Updated 3 weeks ago
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆106Updated last year
- Latent Large Language Models☆18Updated 8 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆60Updated 9 months ago