J

jax

🤗 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/google/grain Library for reading and processing ML training data.

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

Instruction Following Agents with Multimodal Transforemrs

This is a Jax implementation for the InstructRL method.