jax-implementation

There are 4 repositories under jax-implementation topic.

  • tinkoff-ai/sac-rnd

    Official implementation for "Anti-Exploration by Random Network Distillation", ICML 2023

    Language:Python52305

  • 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:

    Language:Jupyter Notebook5301

  • VachanVY/diffusion-transformer

    Pytorch and JAX Implementation of Scalable Diffusion Models with Transformers | Diffusion Transformers in Pytorch and JAX

    Language:Python4200

  • 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 Notebook1100