Awesome Open Source

Programming Languages

Search results for jax tpu

10 search results found

Tpu Starter ⭐ 381

Everything you want to know about Google Cloud TPU

Jaxfluids ⭐ 203

Differentiable Fluid Dynamics Package

Stylegan2 Flax Tpu ⭐ 123

🖼 Training StyleGAN2 on TPUs in JAX

Efficientnet Jax ⭐ 121

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

HomebrewNLP in JAX flavour for maintable TPU-Training

🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX

Big_vision_experiments ⭐ 19

Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.

Polyglot Jax Inference ⭐ 8

TPU에서 한국어용 LLM 추론을 위한 Jax/Flax 구현체입니다.

Starcoder Jax ⭐ 6

a Jax/Flax inference code of StarCoder

Jax_tpu_docker ⭐ 5

Provides a Docker image for running JAX on TPU VMs. This repository contains the necessary files to build the Docker image and run JAX on TPUs with ease, making it easy for researchers to take advantage of the power of TPUs for their machine learning projects.

1-10 of 10 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.