noahfarr / rlx
A reinforcement learning framework based on MLX.
☆233Updated last month
Alternatives and similar repositories for rlx:
Users that are interested in rlx are comparing it to the libraries listed below
- Fast parallel LLM inference for MLX☆179Updated 9 months ago
- run paligemma in real time☆131Updated 10 months ago
- The history files when recording human interaction while solving ARC tasks☆106Updated this week
- ☆107Updated 3 months ago
- Explore a simple example of utilizing MLX for RAG application running locally on your Apple Silicon device.☆168Updated last year
- An mlx project to train a base model on your whatsapp chats using (Q)Lora finetuning☆166Updated last year
- smol models are fun too☆92Updated 5 months ago
- Simple Transformer in Jax☆136Updated 9 months ago
- Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊☆121Updated last week
- General multi-task deep RL Agent☆179Updated 10 months ago
- Start a server from the MLX library.☆182Updated 8 months ago
- procedural reasoning datasets☆559Updated this week
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆105Updated last year
- Efficient framework-agnostic data loading☆420Updated 2 weeks ago
- Fast bare-bones BPE for modern tokenizer training☆152Updated 2 weeks ago
- Gradient descent is cool and all, but what if we could delete it?☆103Updated this week
- FastMLX is a high performance production ready API to host MLX models.☆289Updated last month
- Train your own SOTA deductive reasoning model☆83Updated last month
- smolLM with Entropix sampler on pytorch☆151Updated 5 months ago
- ☆94Updated 6 months ago
- run embeddings in MLX☆86Updated 6 months ago
- Code to train and evaluate Neural Attention Memory Models to obtain universally-applicable memory systems for transformers.☆302Updated 5 months ago
- Minimal example scripts of the Hugging Face Trainer, focused on staying under 150 lines☆198Updated 11 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆80Updated last month
- Implementation of Diffusion Transformer (DiT) in JAX☆270Updated 10 months ago
- ☆86Updated this week
- This repository contain the simple llama3 implementation in pure jax.☆63Updated 2 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO☆57Updated 6 months ago
- Benchmark of Apple MLX operations on all Apple Silicon chips (GPU, CPU) + MPS and CUDA.☆171Updated last week
- ☆117Updated 8 months ago