sdan / vlm-gymLinks
RL gym for vision language models written in JAX
☆135Updated last month
Alternatives and similar repositories for vlm-gym
Users that are interested in vlm-gym are comparing it to the libraries listed below
Sorting:
- look how they massacred my boy☆63Updated last year
- Cerule - A Tiny Mighty Vision Model☆68Updated last month
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆108Updated 9 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆59Updated 2 months ago
- Repository to create traveling waves integrate special information through time☆56Updated 4 months ago
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆62Updated last year
- Lego for GRPO☆30Updated 7 months ago
- ☆68Updated 7 months ago
- ☆47Updated 7 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆84Updated 4 months ago
- Plotting (entropy, varentropy) for small LMs☆99Updated 7 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated last year
- j1-micro (1.7B) & j1-nano (600M) are absurdly tiny but mighty reward models.☆99Updated 5 months ago
- ☆14Updated 8 months ago
- An automated tool for discovering insights from research papaer corpora☆137Updated last year
- Sparse autoencoders for Contra text embedding models☆25Updated last year
- ☆40Updated last year
- A collection of lightweight interpretability scripts to understand how LLMs think☆74Updated this week
- OpenCoconut implements a latent reasoning paradigm where we generate thoughts before decoding.☆174Updated 11 months ago
- Official CLI and Python SDK for Prime Intellect - access GPU compute, remote sandboxes, RL environments, and distributed training infrast…☆121Updated this week
- ☆36Updated 4 months ago
- smolLM with Entropix sampler on pytorch☆149Updated last year
- Train your own SOTA deductive reasoning model☆107Updated 9 months ago
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…☆122Updated 2 months ago
- Project code for training LLMs to write better unit tests + code☆21Updated 7 months ago
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆77Updated 10 months ago
- ☆136Updated 9 months ago
- [ACL 2024] Do Large Language Models Latently Perform Multi-Hop Reasoning?☆84Updated 9 months ago
- ☆89Updated last year
- ☆86Updated last year