NyxAI

stylegan2-flax-tpu

🖼 Training StyleGAN2 on TPUs in JAX

.github

diffusers

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

nyx-ai.github.io

transformers

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