PyPI page
Home page
Author:
Abhivansh Gupta, Vidushee Vats, Ram Mohan Rao Kadiyala, Vinay Kumar Sankarapu
License:
Lexsi Labs Source Available License (LSAL) v1.1
Summary:
AgentTune: build agentic workflows, then train, evaluate, distill, and self-heal them through one trajectory schema. A TRL-backed alignment + fine-tuning library (full RL coverage — DPO, PPO, GRPO) with an agentic integration spine on top: agent-design / memory-design / harness-design pillars, a Project lifecycle, and a FastAPI service.
Latest version:
1.1.3
Required dependencies:
accelerate
|
asteval
|
beautifulsoup4
|
bert-score
|
bitsandbytes
|
chromadb
|
click
|
codebleu
|
colorama
|
compressed-tensors
|
datasets
|
evaluate
|
groq
|
httpx
|
huggingface-hub
|
jsonschema
|
langchain
|
langchain-community
|
langchain-text-splitters
|
langgraph
|
litellm
|
lm-eval
|
matplotlib
|
networkx
|
nltk
|
numpy
|
openenv
|
pandas
|
pdfplumber
|
peft
|
pillow
|
psutil
|
psycopg2-binary
|
pyarrow
|
pydantic
|
python-docx
|
python-pptx
|
pyyaml
|
requests
|
rich
|
rouge-score
|
scikit-learn
|
sentence-transformers
|
tensorboard
|
torch
|
torchao
|
torchvision
|
tqdm
|
transformers
|
trl
|
typer
|
wandb
|
websockets
Optional dependencies:
bandit
|
black
|
fastapi
|
flash-attn
|
hypothesis
|
isort
|
mkdocs
|
mkdocs-material
|
mypy
|
pre-commit
|
pymdown-extensions
|
pytest
|
pytest-asyncio
|
pytest-cov
|
pytest-mock
|
pytest-timeout
|
pytest-xdist
|
ruff
|
trl
|
uvicorn
|
websockets
Downloads last day:
9
Downloads last week:
50
Downloads last month:
1,617
7-day smoothing