TristanBilot / mlx-GCNLinks
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:
- A collection of optimizers for MLX☆36Updated last week
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆29Updated 11 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 last year
- Clean RL implementation using MLX☆32Updated last year
- Graph Neural Network library made for Apple Silicon☆191Updated 8 months ago
- FMS Model Optimizer is a framework for developing reduced precision neural network models.☆20Updated this week
- ☆14Updated last year
- A fast minimalistic implementation of guided generation on Apple Silicon using Outlines and MLX☆53Updated last year
- OpenPipe Reinforcement Learning Experiments☆25Updated 2 months ago
- ☆27Updated 10 months ago
- Training hybrid models for dummies.☆21Updated 4 months ago
- Very basic framework for composable parameterized large language model (Q)LoRA / (Q)Dora fine-tuning using mlx, mlx_lm, and OgbujiPT.☆40Updated this week
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 2 months ago
- Implementation of SelfExtend from the paper "LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning" from Pytorch and Zeta☆13Updated 6 months ago
- A python command-line tool to download & manage MLX AI models from Hugging Face.☆17Updated 9 months ago
- Minimal, clean code implementation of RAG with mlx using gguf model weights☆50Updated last year
- The application is a end-user training and evaluation system for standard knowledge graph embedding models. It was developed to optimise …☆18Updated last week
- ☆66Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- A package for defining deep learning models using categorical algebraic expressions.☆60Updated 10 months ago
- ☆130Updated 9 months ago
- A repository for research on medium sized language models.☆76Updated last year
- Artificial Life simulations☆52Updated 2 months ago
- BH hackathon☆14Updated last year
- alternative way to calculating self attention☆18Updated last year
- Latent Large Language Models☆18Updated 9 months ago
- Structured Generation Evals☆12Updated 8 months ago
- Implementation of Spectral State Space Models☆16Updated last year
- 🧮 Algebraic Positional Encodings.☆13Updated 4 months ago