TheDuckAI / prm
☆12Updated 2 weeks ago
Alternatives and similar repositories for prm:
Users that are interested in prm are comparing it to the libraries listed below
- Minimal but scalable implementation of large language models in JAX☆28Updated 2 months ago
- ☆20Updated 4 months ago
- ☆76Updated 6 months ago
- ☆28Updated last month
- Sparse and discrete interpretability tool for neural networks☆59Updated 11 months ago
- ☆34Updated 2 years ago
- ☆80Updated 6 months ago
- Learn online intrinsic rewards from LLM feedback☆34Updated last month
- Scaling scaling laws with board games.☆45Updated last year
- A library to create and manage configuration files, especially for machine learning projects.☆76Updated 2 years ago
- ☆30Updated 2 months ago
- Interpreting how transformers simulate agents performing RL tasks☆77Updated last year
- [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)☆20Updated 5 months ago
- Code supporting the preprint "Training Language Models on Synthetic Edit Sequences Improves Code Synthesis" (Piterbarg, Pinto, Fergus)☆10Updated 3 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated last month
- Advantage Leftover Lunch Reinforcement Learning (A-LoL RL): Improving Language Models with Advantage-based Offline Policy Gradients☆26Updated 4 months ago
- Implementation of Direct Preference Optimization☆15Updated last year
- ☆15Updated last year
- This is a pip package implementing Reinforcement Learning algorithms in non-stationary environments supported by the OpenAI Gym toolkit.☆12Updated 7 months ago
- Minimum Description Length probing for neural network representations☆18Updated this week
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆18Updated 7 months ago
- Simple and efficient pytorch-native transformer training and inference (batched)☆67Updated 9 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 8 months ago
- Repository for the code of the "PPL-MCTS: Constrained Textual Generation Through Discriminator-Guided Decoding" paper, NAACL'22☆65Updated 2 years ago
- ☆51Updated 8 months ago
- ☆25Updated 9 months ago
- ☆18Updated 8 months ago
- A repo for RLHF training and BoN over LLMs, with support for reward model ensembles.☆35Updated 2 weeks ago
- BASALT Benchmark datasets, evaluation code and agent training example.☆20Updated last year
- Official repository for our paper, Transformers Learn Higher-Order Optimization Methods for In-Context Learning: A Study with Linear Mode…☆15Updated 2 months ago