LICENSE
README.md
pyproject.toml
imitateai/__init__.py
imitateai/modeling.py
imitateai/refinement.py
imitateai.egg-info/PKG-INFO
imitateai.egg-info/SOURCES.txt
imitateai.egg-info/dependency_links.txt
imitateai.egg-info/requires.txt
imitateai.egg-info/top_level.txt
imitateai/common/__init__.py
imitateai/common/utils.py
imitateai/data/__init__.py
imitateai/data/demonstration.py
imitateai/examples/01_infer_reward_model.py
imitateai/examples/02_recursive_refinement.py
imitateai/irl/__init__.py
imitateai/irl/base.py
imitateai/irl/gail.py
imitateai/irl/maxent.py
imitateai/models/__init__.py
imitateai/models/reward_model.py
imitateai/rl/__init__.py
imitateai/rl/grpo_trainer.py
imitateai/rl/ppo_trainer.py
imitateai/tests/__init__.py
imitateai/tests/test_modeling.py
imitateai/tests/test_refinement.py