TristanBilot / mlx-GCN
MLX implementation of GCN, with benchmark on MPS, CUDA and CPU (M1 Pro, M2 Ultra, M3 Max).
☆20Updated 9 months ago
Related projects: ⓘ
- Clean RL implementation using MLX☆26Updated 6 months ago
- Graph Neural Network library made for Apple Silicon☆159Updated 3 weeks ago
- mlx image models for Apple Silicon machines☆58Updated 4 months ago
- MLX implementation of xLSTM model by Beck et al. (2024)☆21Updated 3 months ago
- Minimal, clean code implementation of RAG with mlx using gguf model weights☆40Updated 4 months ago
- BH hackathon☆14Updated 5 months ago
- A python command-line tool to download & manage MLX AI models from Hugging Face.☆15Updated 3 weeks ago
- Gpu benchmark☆35Updated 2 weeks ago
- A fast minimalistic implementation of guided generation on Apple Silicon using Outlines and MLX☆42Updated 7 months ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆43Updated 3 weeks ago
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆21Updated 2 months ago
- MLX Transformers is a library that provides model implementation in MLX. It uses a similar model interface as HuggingFace Transformers an…☆47Updated last month
- Implementation of Spectral State Space Models☆16Updated 6 months ago
- run embeddings in MLX☆68Updated last month
- Benchmark of Apple MLX operations on all Apple Silicon chips (GPU, CPU) + MPS and CUDA.☆113Updated 2 weeks ago
- ☆19Updated last month
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆96Updated 7 months ago
- Very basic framework for parameterized large language model (Q)LoRa fine-tuning using mlx, mlx_lm, and OgbujiPT. Architecture for system…☆32Updated last month
- Tools to make language models a bit easier to use☆22Updated last week
- ☆14Updated 4 months ago
- alternative way to calculating self attention☆18Updated 3 months ago
- A simple example of VAEs with KANs☆12Updated 4 months ago
- ☆37Updated last month
- Graph neural networks in JAX.☆66Updated 3 months ago
- Implementation of SelfExtend from the paper "LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning" from Pytorch and Zeta☆13Updated last week
- Collection of autoregressive model implementation☆62Updated 2 weeks ago
- ☆27Updated 2 months ago
- Triton Implementation of HyperAttention Algorithm☆46Updated 9 months ago
- Custom kernels in Triton language for accelerating LLMs☆14Updated 5 months ago
- ☆17Updated 4 months ago