TristanBilot / mlx-GCNLinks
MLX implementation of GCN, with benchmark on MPS, CUDA and CPU (M1 Pro, M2 Ultra, M3 Max).
☆23Updated 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:
- The application is a end-user training and evaluation system for standard knowledge graph embedding models. It was developed to optimise …☆18Updated last month
- Training hybrid models for dummies.☆25Updated 6 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 4 months ago
- An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO☆30Updated this week
- FMS Model Optimizer is a framework for developing reduced precision neural network models.☆20Updated this week
- Package of useful sampling algorithms written in MLX.☆17Updated last year
- Clean RL implementation using MLX☆32Updated last year
- Jax like function transformation engine but micro, microjax☆33Updated 8 months ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆71Updated 2 weeks ago
- A collection of optimizers for MLX☆37Updated this week
- Tasks and tutorials using Graphore's IPU with Hugging Face. Originally at https://github.com/gradient-ai/Graphcore-HuggingFace☆16Updated last year
- Implementation of Spectral State Space Models☆16Updated last year
- ☆27Updated last year
- Code for the paper "Function-Space Learning Rates"☆20Updated last month
- BH hackathon☆14Updated last year
- Collection of LLM completions for reasoning-gym task datasets☆26Updated 2 weeks ago
- Graph Neural Network library made for Apple Silicon☆191Updated 9 months ago
- alternative way to calculating self attention☆18Updated last year
- https://x.com/BlinkDL_AI/status/1884768989743882276☆28Updated 2 months ago
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆110Updated last year
- Very minimal (and stateless) agent framework☆44Updated 6 months ago
- Adaptive Parallel PDF Parsing and Resource Scaling Engine☆48Updated last month
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆32Updated last year
- Implementation of SelfExtend from the paper "LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning" from Pytorch and Zeta☆13Updated 8 months ago
- mlx image models for Apple Silicon machines☆82Updated 3 months ago
- Repository to create traveling waves integrate special information through time☆53Updated 4 months ago
- A simple example of VAEs with KANs☆12Updated last year
- OpenPipe Reinforcement Learning Experiments☆27Updated 4 months ago
- Einsum-like high-level array sharding API for JAX☆35Updated last year
- Simple repository for training small reasoning models☆33Updated 5 months ago