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
- Roberta Question Answering using MLX.☆23Updated last year
- Implementation of Spectral State Space Models☆16Updated 11 months ago
- A collection of optimizers for MLX☆29Updated this week
- Clean RL implementation using MLX☆28Updated 10 months ago
- MLX implementation of xLSTM model by Beck et al. (2024)☆26Updated 7 months ago
- ☆14Updated 8 months ago
- never forget anything again! combine AI and intelligent tooling for a local knowledge base to track catalogue, annotate, and plan for you…☆36Updated 8 months ago
- Graph Neural Network library made for Apple Silicon☆179Updated 3 months ago
- Minimal, clean code implementation of RAG with mlx using gguf model weights☆46Updated 9 months ago
- BH hackathon☆14Updated 9 months ago
- A fast minimalistic implementation of guided generation on Apple Silicon using Outlines and MLX☆51Updated 11 months ago
- An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO☆27Updated this week
- A python command-line tool to download & manage MLX AI models from Hugging Face.☆17Updated 5 months ago
- Very basic framework for parameterized large language model (Q)LoRA / (Q)Dora fine-tuning using mlx, mlx_lm, and OgbujiPT. Architecture …☆36Updated this week
- ☆53Updated 2 months ago
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆23Updated 7 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆26Updated 2 weeks ago
- ☆27Updated 6 months ago
- Training hybrid models for dummies.☆18Updated 2 weeks ago
- ☆52Updated 9 months ago
- Graph neural networks in JAX.☆67Updated 7 months ago
- JAX implementation of the Mistral 7b v0.2 model☆35Updated 6 months ago
- alternative way to calculating self attention☆18Updated 8 months ago
- Jax like function transformation engine but micro, microjax☆30Updated 3 months ago
- Very minimal (and stateless) agent framework☆41Updated 2 weeks ago
- ☆18Updated 2 months ago
- Using multiple LLMs for ensemble Forecasting☆16Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆18Updated last week
- Collection of autoregressive model implementation☆78Updated 3 weeks ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 3 weeks ago