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☆33Updated last year
- Graph Neural Network library made for Apple Silicon☆202Updated 2 months ago
- A simple example of VAEs with KANs☆11Updated last year
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆74Updated 4 months ago
- An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO☆29Updated this week
- A collection of optimizers for MLX☆54Updated 3 weeks ago
- FMS Model Optimizer is a framework for developing reduced precision neural network models.☆20Updated this week
- 🆕 Benchmarking Goal-Oriented Software Engineering☆35Updated last week
- Jax like function transformation engine but micro, microjax☆33Updated last year
- Very minimal (and stateless) agent framework☆45Updated 10 months ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆43Updated last year
- OpenPipe Reinforcement Learning Experiments☆32Updated 8 months ago
- Because we don't want a jupyter notebook mess...☆61Updated 5 months ago
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆114Updated last year
- A Python Library for Learning Non-Euclidean Representations☆67Updated 3 months ago
- Implementation of SelfExtend from the paper "LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning" from Pytorch and Zeta☆12Updated last year
- never forget anything again! combine AI and intelligent tooling for a local knowledge base to track catalogue, annotate, and plan for you…☆38Updated last year
- Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊☆134Updated last month
- A fast minimalistic implementation of guided generation on Apple Silicon using Outlines and MLX☆57Updated last year
- ☆36Updated 3 months ago
- nanoGPT using Equinox☆13Updated 2 years ago
- Tensor-Slayer : Manipulate weights and tensors of LLMs to achieve performance upgrades and introduce a novel inferenceless mechanistic in…☆26Updated 5 months ago
- aesthetic tensor visualiser☆27Updated 6 months ago
- Neural Networks for JAX☆84Updated last year
- Code for the paper "Function-Space Learning Rates"☆23Updated 5 months ago
- Training hybrid models for dummies.☆29Updated 2 weeks ago
- alternative way to calculating self attention☆18Updated last year
- Artificial Life simulations☆54Updated 8 months ago
- MLX implementation of xLSTM model by Beck et al. (2024)☆29Updated last year
- The application is a end-user training and evaluation system for standard knowledge graph embedding models. It was developed to optimise …☆18Updated 5 months ago