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:
- Clean RL implementation using MLX☆32Updated last year
- FMS Model Optimizer is a framework for developing reduced precision neural network models.☆20Updated this week
- A collection of optimizers for MLX☆36Updated last month
- KAN (Kolmogorov–Arnold Networks) in the MLX framework for Apple Silicon☆18Updated last week
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 3 months ago
- BH hackathon☆14Updated last year
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆30Updated last year
- A python command-line tool to download & manage MLX AI models from Hugging Face.☆17Updated 10 months ago
- Graph Neural Network library made for Apple Silicon☆193Updated 8 months ago
- The official evaluation suite and dynamic data release for MixEval.☆11Updated 9 months ago
- ☆19Updated 5 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆80Updated last month
- Matrix (Multi-Agent daTa geneRation Infra and eXperimentation framework) is a versatile engine for multi-agent conversational data genera…☆71Updated this week
- Collection of LLM completions for reasoning-gym task datasets☆24Updated last month
- ☆13Updated last week
- The application is a end-user training and evaluation system for standard knowledge graph embedding models. It was developed to optimise …☆18Updated 3 weeks ago
- A framework for pitting LLMs against each other in an evolving library of games ⚔☆32Updated 2 months ago
- MLX implementation of xLSTM model by Beck et al. (2024)☆28Updated last year
- Implementation of SelfExtend from the paper "LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning" from Pytorch and Zeta☆13Updated 7 months ago
- OpenPipe Reinforcement Learning Experiments☆25Updated 3 months ago
- Graph neural networks in JAX.☆67Updated last year
- Latent Large Language Models☆18Updated 10 months ago
- Very minimal (and stateless) agent framework☆44Updated 5 months ago
- run embeddings in MLX☆90Updated 9 months ago
- An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO☆29Updated this week
- ☆27Updated 11 months ago
- Using multiple LLMs for ensemble Forecasting☆16Updated last year
- ☆61Updated 7 months ago
- Repository to create traveling waves integrate special information through time☆53Updated 3 months ago
- Implementation of Spectral State Space Models☆16Updated last year