Adaptive Stress Testing for Robust AI & Reinforcement Learning (ASTRA-RL)¶
Warning
This documentation (and project) is a work in progress. If you have any questions or suggestions, please feel free to open an issue on the GitHub repository
ASTRA-RL is a python toolbox for training and evaluating language models and generative AI systems that use textual inputs. It provides a set of tools for training, evaluating, and analyzing language models, with a focus on applying reinforcement learning based refinement techniques to improve evaluator model performance.
Installation¶
To install the ASTRA-RL toolbox, you can use pip. The package is available on PyPI, so you can install it directly from there.
pip install astra-rl
You can then import the library in your Python code:
import astra_rl
# or
import astra_rl as astral
Using the Documentation¶
This documentation is structured to help you quickly find the information you need. The main sections include:
- Getting Started: An introduction to the toolbox, including installation and basic usage.
- Tutorials: Step-by-step guides to help you understand how to use the toolbox effectively.
- API Reference: Detailed information about the classes, functions, and modules available in the