F

flax

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

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

🔧🔗github hf diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

🕸️🔗huggingface.co/docs/diffusers

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

Instruction Following Agents with Multimodal Transforemrs

This is a Jax implementation for the InstructRL method.