Automatically take good care of your preemptible TPUs
☆37May 15, 2023Updated 2 years ago
Alternatives and similar repositories for tpucare
Users that are interested in tpucare are comparing it to the libraries listed below
Sorting:
- Implementation of PSGD optimizer in JAX☆35Dec 31, 2024Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated 7 months ago
- ☆31Mar 9, 2026Updated last week
- PyTorch interface for TrueGrad Optimizers☆43Aug 8, 2023Updated 2 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Aug 4, 2022Updated 3 years ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Jan 20, 2024Updated 2 years ago
- DiT (training + flow matching) in Jax☆11Jan 5, 2025Updated last year
- ☆16Dec 30, 2024Updated last year
- ☆21Jan 23, 2024Updated 2 years ago
- Autoresize Textarea for Svelte apps using proxy element.☆29Sep 12, 2023Updated 2 years ago
- Exploring finetuning public checkpoints on filter 8K sequences on Pile☆115Mar 22, 2023Updated 3 years ago
- Script and models for clustering LAION-400m CLIP embeddings.☆26Jan 10, 2022Updated 4 years ago
- Train vision models using JAX and 🤗 transformers☆101Dec 14, 2025Updated 3 months ago
- JAX implementation ViT-VQGAN☆82Sep 21, 2022Updated 3 years ago
- Optimized library for large-scale extraction of frames and audio from video.☆201Sep 11, 2023Updated 2 years ago
- ☆33Nov 4, 2024Updated last year
- Implementation of N-Grammer in Flax☆17Nov 3, 2022Updated 3 years ago
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- (EasyDel Former) is a utility library designed to simplify and enhance the development in JAX☆29Mar 12, 2026Updated last week
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 3 months ago
- This project provides a data set with bounding boxes, body poses, 3D face meshes & captions of people from our LAION-2.2B. Additionally i…☆14Jan 2, 2022Updated 4 years ago
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated last year
- Synthetic Data Generation for Evaluation☆13Feb 21, 2025Updated last year
- Amos optimizer with JEstimator lib.☆82May 15, 2024Updated last year
- Efficient optimizers☆294Updated this week
- ☆19Dec 4, 2025Updated 3 months ago
- ☆13Apr 25, 2024Updated last year
- Tool for generating pictures using mathematical formulas.☆43Nov 8, 2021Updated 4 years ago
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆17Aug 1, 2022Updated 3 years ago
- ☆13Jun 22, 2025Updated 9 months ago
- A Constraint Solver☆12Dec 4, 2015Updated 10 years ago
- Easily turn large sets of audio urls to an audio dataset.☆21Dec 27, 2022Updated 3 years ago
- Hex encode & decode a string, right from your terminal.☆10Jan 5, 2023Updated 3 years ago
- ☆15May 11, 2025Updated 10 months ago
- Wallpaper downloader and rotator that uses Flickr and Webshots.☆30Jul 12, 2014Updated 11 years ago
- An implementation of DecorrelatedBN by tensorflow☆13Jun 30, 2022Updated 3 years ago
- ☆12Apr 26, 2024Updated last year
- ☆10Apr 5, 2024Updated last year