Skip to content

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