We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add jax support for RLHF on TPUs.
No response
The text was updated successfully, but these errors were encountered:
We should, rather than just add JAX support, add T5X support so it can be used with models that were trained via T5X.
Sorry, something went wrong.
can it be trained via T5X now ?
T5X support is something we're planning for late 2023 at the earliest.
No branches or pull requests
馃殌 The feature, motivation, and pitch
Add jax support for RLHF on TPUs.
Alternatives
No response
Additional context
No response
The text was updated successfully, but these errors were encountered: