transformersTransformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
ctranslate2Fast inference engine for Transformer models
inspect-aiFramework for large language model evaluations
torchtextText utilities, models, transforms, and datasets for PyTorch.
torch-model-archiverTorch Model Archiver is used for creating archives of trained neural net models that can be consumed by TorchServe inference
snowballstemmerThis package provides 32 stemmers for 30 languages generated from Snowball algorithms.
stable-baselines3Pytorch version of Stable Baselines, implementations of reinforcement learning algorithms.
trl
Summary: Train transformer language models with reinforcement learning.
Author: Leandro von Werra <leandro.vonwerra@gmail.com>