sdan / vlm-gymLinks
Simple RL gym for vision models in pure JAX
☆93Updated this week
Alternatives and similar repositories for vlm-gym
Users that are interested in vlm-gym are comparing it to the libraries listed below
Sorting:
- ☆14Updated 6 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated 11 months ago
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆62Updated 11 months ago
- Sparse autoencoders for Contra text embedding models☆25Updated last year
- look how they massacred my boy☆63Updated last year
- The Prime Intellect CLI provides a powerful command-line interface for managing GPU resources across various providers☆98Updated this week
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆84Updated last month
- ☆68Updated 4 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆106Updated 7 months ago
- Cerule - A Tiny Mighty Vision Model☆67Updated last year
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆72Updated 8 months ago
- Plotting (entropy, varentropy) for small LMs☆98Updated 5 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆55Updated 8 months ago
- An automated tool for discovering insights from research papaer corpora☆138Updated last year
- SIMD quantization kernels☆87Updated last month
- Train your own SOTA deductive reasoning model☆108Updated 7 months ago
- Training-Ready RL Environments + Evals☆128Updated this week
- ☆45Updated 4 months ago
- ☆21Updated 9 months ago
- Simple Transformer in Jax☆139Updated last year
- train entropix like a champ!☆20Updated last year
- papers.day☆90Updated last year
- ☆89Updated last year
- ☆40Updated last year
- ☆36Updated 2 months ago
- Repository to create traveling waves integrate special information through time☆55Updated 2 months ago
- EXO Gym is an open-source Python toolkit that facilitates distributed AI research.☆78Updated last month
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆112Updated last year
- ☆96Updated last year
- a highly efficient compression algorithm for the n1 implant (neuralink's compression challenge)☆46Updated last year