New features and optimizations
- TRT-LLM support in the RLHF pipeline. This adds significant speedup(6.96x) over our previous implementation.
Container
Please refer to the Dockerfile which contains all the dependencies needed to run TRT-LLM and RLHF.
Learn More
For more information see our Paper and Usage Guide.