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:
- Graph Neural Network library made for Apple Silicon☆201Updated 2 months ago
- Clean RL implementation using MLX☆33Updated last year
- You should use PySR to find scaling laws. Here's an example.☆33Updated 2 years ago
- FMS Model Optimizer is a framework for developing reduced precision neural network models.☆20Updated this week
- Graph neural networks in JAX.☆68Updated last year
- A collection of optimizers for MLX☆53Updated last week
- Jax like function transformation engine but micro, microjax☆33Updated last year
- MLX implementation of xLSTM model by Beck et al. (2024)☆29Updated last year
- An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO☆29Updated this week
- Because we don't want a jupyter notebook mess...☆61Updated 4 months ago
- A fast minimalistic implementation of guided generation on Apple Silicon using Outlines and MLX☆57Updated last year
- Very minimal (and stateless) agent framework☆45Updated 9 months ago
- Einsum-like high-level array sharding API for JAX☆35Updated last year
- JAX implementation of the Mistral 7b v0.2 model☆35Updated last year
- ☆28Updated last year
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆112Updated last year
- An introduction to LLM Sampling☆79Updated 10 months ago
- Code for the paper "Function-Space Learning Rates"☆23Updated 4 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆84Updated 2 months ago
- ☆60Updated 3 years ago
- Training hybrid models for dummies.☆27Updated last week
- A simple example of VAEs with KANs☆12Updated last year
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆74Updated 3 months ago
- https://x.com/BlinkDL_AI/status/1884768989743882276☆28Updated 5 months ago
- Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊☆132Updated this week
- A collection of lightweight interpretability scripts to understand how LLMs think☆61Updated this week
- Minimal, clean code implementation of RAG with mlx using gguf model weights☆52Updated last year
- OpenPipe Reinforcement Learning Experiments☆31Updated 7 months ago
- ☆45Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated 2 years ago