|
|
Python |
3 |
Single thread version for training BigGAN with TPU. |
Dec 02, 2020 |
|
|
Python |
4 |
Google Cloud TPU Utilization Bar for Training Models |
Jun 08, 2022 |
|
|
Python |
60 |
CLOOB training (JAX) and inference (JAX and PyTorch) |
Aug 23, 2022 |
|
|
Python |
9 |
Distributed version (multiple-process) for training BigGAN with TPU. |
Sep 16, 2022 |
|
|
Python |
2 |
Parallel version (multiple-thread) for training BigGAN with TPU. |
Jun 14, 2022 |
|
|
Python |
175 |
Minimal library to train LLMs on TPU in JAX with pjit(). |
Apr 24, 2023 |
|
|
Jupyter Notebook |
4 |
stable diffusion text to image cuda and jax tpu model repo. |
Mar 20, 2023 |
|
|
Python |
4 |
Model training utilities in JAX. |
Mar 14, 2023 |
|
|
Python |
68 |
A functional training loops library for JAX |
May 25, 2023 |
|
|
C++ |
2 |
Serve compiled XLA from Nx, JAX, TensorFlow, etc. on Cloud TPU VMs |
Nov 14, 2022 |
|
|
Python |
7 |
JAX implementation of LLaMA, aiming to train LLaMA on Google Cloud TPU |
Jun 15, 2023 |
|
|
Python |
37 |
Pre-training script for BART in JAX/Flax |
Aug 11, 2022 |
|
|
Python |
63 |
🖼 Training StyleGAN2 on TPUs in JAX |
Jul 21, 2022 |
|
|
Dockerfile |
17 |
A collection of single-host tests for Atomic Host |
Aug 05, 2022 |
|
|
Python |
25 |
Shared code for training sentence embeddings with Flax / JAX |
Feb 14, 2023 |
|
|
Python |
1504 |
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU. |
Sep 11, 2022 |
|
|
None |
2 |
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU. |
Aug 03, 2023 |
|
|
Jupyter Notebook |
11 |
TPU use in single line in colab using tf2 package. |
Dec 28, 2021 |
|
|
Python |
11 |
A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for … |
Jul 24, 2022 |
|
|
Shell |
19 |
Host Linux Training public documents |
Feb 17, 2022 |
|
|
Jupyter Notebook |
125 |
Colab/Jupyter tutorials about training TensorFlow models for Edge TPU, and other tutorials |
May 05, 2023 |
|
|
Jupyter Notebook |
112 |
CLU lets you write beautiful training loops in JAX. |
Aug 08, 2022 |
|
|
None |
4 |
A curated space for the JAX Single Cell Community of Practice |
Feb 07, 2020 |
|
|
Shell |
4 |
Cassandra multi node setup script for single host |
Feb 14, 2021 |
|
|
Go |
4 |
Tiny single host/IP DNS server |
Jun 16, 2022 |
|
|
Smarty |
4 |
ISAR Layer for Coral TPU |
Mar 04, 2022 |
|
|
Jupyter Notebook |
8 |
Repository to host code for princeton public library python training |
Feb 07, 2022 |
|
|
TypeScript |
3209 |
Host single page apps with GitHub Pages |
Sep 04, 2022 |
|
|
Go |
2 |
A simple single host Golang reverse proxy |
Jul 20, 2016 |
|
|
HTML |
2 |
Repo and webpage to host GCAM training material |
Jan 13, 2024 |
|
|
Shell |
2 |
Monit setup for docker single-host monitoring (host, docker daemon and docker containers) with UI |
Oct 21, 2023 |
|
|
Python |
8 |
generative language model training on top of the JAX and Huggingface 🤗 |
Jul 13, 2022 |
|
|
Jupyter Notebook |
13 |
TPU support for the fastai library |
Aug 08, 2022 |
|
|
Java |
34 |
JAX-WS API (Eclipse Project for JAX-WS) |
Jul 29, 2022 |
|
|
Python |
9 |
DataCite-based data model for InvenioRDM flavour. |
Mar 17, 2022 |
|
|
Emacs Lisp |
8 |
Additional flavour for your Emacs Lisp programming |
Jan 09, 2022 |
|
|
HTML |
3 |
Materials for training provision on single-cell RNA-seq |
Apr 13, 2021 |
|
|
Python |
2 |
The board game Go implemented in JAX for fast game processing and machine learning training. |
Feb 04, 2023 |
|
|
Python |
35 |
Tools for JAX |
Apr 02, 2023 |
|
|
Python |
3 |
A minimal JAX re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training |
Oct 27, 2023 |
|
|
Shell |
5 |
Setup for a K8s home lab running on a single host |
Aug 25, 2022 |
|
|
Verilog |
37 |
IC implementation of Systolic Array for TPU |
Apr 20, 2023 |
|
|
Python |
19 |
ROS package for the Google Edge TPU |
Jun 21, 2022 |
|
|
None |
69 |
Simple, single-file deploy Django to an Ubuntu host |
Jan 21, 2022 |
|
|
HTML |
31 |
Single web page applications for customizing and training GPT-3 |
Apr 09, 2023 |
|
|
Ruby |
34 |
Core library for the Ruby flavour of Cucumber |
Nov 16, 2021 |
|
|
CSS |
37 |
Warhammer-flavour CSS for Battlescribe's HTML roster output |
May 20, 2023 |
|
|
Python |
268 |
PIX is an image processing library in JAX, for JAX. |
Aug 15, 2022 |
|
|
Python |
7 |
KLU solver for JAX |
Jul 13, 2022 |
|
|
Java |
2 |
Documentation for JAX-RS |
Oct 25, 2020 |