jax

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

huggingface.co/transformers https://github.com/huggingface/transformers

🔧🔗https://github.com/forhaoliu/instructrl

Instruction Following Agents with Multimodal Transforemrs

This is a Jax implementation for the InstructRL method.