GitHub topics: jax-implementation
VachanVY/diffusion-transformer
Pytorch and JAX Implementation of Scalable Diffusion Models with Transformers | Diffusion Transformers in Pytorch and JAX
Language: Python - Size: 56.5 MB - Last synced at: 10 days ago - Pushed at: 10 days ago - Stars: 7 - Forks: 0

tinkoff-ai/sac-rnd
Official implementation for "Anti-Exploration by Random Network Distillation", ICML 2023
Language: Python - Size: 22.5 KB - Last synced at: almost 2 years ago - Pushed at: over 2 years ago - Stars: 23 - Forks: 1

manhph2211/Image-Generation-App
This repo shows how to implement a simple image generation app that uses conditional VAE, Jax, fastapi, docker, streamlit, heroku, ec2, and cloudflare :smiley:
Language: Jupyter Notebook - Size: 2.75 MB - Last synced at: about 2 years ago - Pushed at: about 2 years ago - Stars: 2 - Forks: 0

RealMlops/Image-Generation
This repo shows how to implement a simple image generation app that uses conditional VAE, Jax, fastapi, docker, streamlit and heroku
Language: Jupyter Notebook - Size: 2.74 MB - Last synced at: about 2 years ago - Pushed at: about 2 years ago - Stars: 1 - Forks: 0
