jax-implementation

There are 0 repository under jax-implementation topic.

tinkoff-ai / sac-rnd
Official implementation for "Anti-Exploration by Random Network Distillation", ICML 2023
deep-reinforcement-learning flax-implementation jax-implementation offline-reinforcement-learning random-network-distillation
Language:Python 54
VachanVY / diffusion-transformer
Pytorch and JAX Implementation of Scalable Diffusion Models with Transformers | Diffusion Transformers in Pytorch and JAX
deep-learning diffusion-models diffusion-transformer jax jax-implementation python pytorch pytorch-implementation transformers
Language:Python 8
manhph2211 / ML-Pipeline
This repo shows how to implement a simple image generation app that uses Jax-Implementation of a conditional VAE, Jax, fastapi, docker, streamlit, heroku, ec2, and cloudflare :smiley:
fastapi heroku image-generation jax streamlit vae play-with-jax generative-models jax-implementation mlops-project mlopsvn ml-deployment
Language:Jupyter Notebook 6
RealMlops / Image-Generation
This repo shows how to implement a simple image generation app that uses conditional VAE, Jax, fastapi, docker, streamlit and heroku
fastapi image-generation jax-implementation play-with-jax streamlit vae
Language:Jupyter Notebook 1

tinkoff-ai / sac-rnd