Agent Gym
Convert any model into a r1-like reasoning hyper-intelligent agent. Leverages TRL, Huggingface, and various other libraries. This is a work in progress. Our goal is to make it easy to train any model into a reasoning agent.
Installation
pip3 install -U agentgym
Usage
from agentgym.r1_pipeline import R1Pipeline, SFTConfig
r1_pipeline = R1Pipeline(sft_model="gpt2", sft_dataset="stanfordnlp/imdb", sft_args=SFTConfig(output_dir="/tmp"))
r1_pipeline.run()
Architecture
The architecture is as follows:
- SFT: Supervised Fine-Tuning
- GRPO: Generative Reinforcement Policy Optimization
-> model -> sft -> grpo -> model
graph TD;
A[model] --> B[sft]
B --> C[grpo]
C --> D[reasoning model]
License
MIT
1 maintainer
Metadata
1 maintainer
Release files for agentgym 0.0.2
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| agentgym-0.0.2.tar.gz | 8.9 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| agentgym-0.0.2-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 17.9 kB
Release files / agentgym-0.0.2.tar.gz
| Download URL | agentgym-0.0.2.tar.gz |
|---|---|
| Size | 8.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
1a62ef68d0173f749dbc1eaf6c7f1e35d38d6ad2c7bb80baa3be0d4df1dcd181
|
|
BLAKE2b-256 checksum How to use checksums |
b92b279cbbe392b6608dbf251032593b7cac7a40f8c8661d1e4206ac3182abe2
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.8.3 CPython/3.12.8 Darwin/23.3.0
|
Release files / agentgym-0.0.2-py3-none-any.whl
| Download URL | agentgym-0.0.2-py3-none-any.whl |
|---|---|
| Size | 9.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
8f987431f6429283e5345bcddf8d9b66afc5940c3c54f3b06639c65e9c7cb022
|
|
BLAKE2b-256 checksum How to use checksums |
670ae4b5639b379da18f947018aef7e31bd3ca0cd81735b1f73d4a77ce229f95
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.8.3 CPython/3.12.8 Darwin/23.3.0
|
