Amos optimizer with JEstimator lib.
☆82May 15, 2024Updated 2 years ago
Alternatives and similar repositories for jestimator
Users that are interested in jestimator are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆16Dec 10, 2022Updated 3 years ago
- JAX implementation of the T5 model: Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer☆24Jun 10, 2023Updated 3 years ago
- ☆14May 4, 2026Updated 3 months ago
- c++ mosestokenizer☆18Mar 13, 2024Updated 2 years ago
- Automatically take good care of your preemptible TPUs☆37May 15, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Code for AAAI 2024 paper: CR-SAM: Curvature Regularized Sharpness-Aware Minimization☆12Nov 29, 2024Updated last year
- ☆16Jul 16, 2024Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 2024Updated 2 years ago
- A minimal Pytorch Implementation of Stochastically Quantized Variational AutoEncoder (SQ-VAE) by Sony☆33Oct 16, 2023Updated 2 years ago
- bumble bee transformer☆14Apr 19, 2021Updated 5 years ago
- Train very large language models in Jax.☆208Oct 21, 2023Updated 2 years ago
- ☆200Aug 10, 2026Updated last week
- [TMLR'23] Contrastive Search Is What You Need For Neural Text Generation☆122Mar 5, 2023Updated 3 years ago
- Official Implementation of "Transferring Inductive Biases Through Knowledge Distillation"☆15Jun 3, 2020Updated 6 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks☆30Jul 20, 2026Updated 3 weeks ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated 2 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Github for the conference paper GLOD-Gaussian Likelihood OOD detector☆16Apr 18, 2022Updated 4 years ago
- ☆13Jan 23, 2017Updated 9 years ago
- Peregrine is a rapid, append-only logging and note-taking app, inspired by @thesephist's Inc.☆24Aug 23, 2025Updated 11 months ago
- ☆18Aug 24, 2024Updated last year
- ☆11Oct 13, 2023Updated 2 years ago
- Plug-and-play Search Interfaces with Pyserini and Hugging Face☆31Aug 5, 2023Updated 3 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- diffGrad: An Optimization Method for Convolutional Neural Networks☆55Oct 12, 2022Updated 3 years ago
- Example codes in the medium post titled "Optuna meets Weights and Biases."☆24Aug 11, 2022Updated 4 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Nov 12, 2020Updated 5 years ago
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆66Dec 16, 2022Updated 3 years ago
- Using Rainbow implementation in Chainer RL for Slime Volleyball Pixel Environment☆23Jun 8, 2020Updated 6 years ago
- An alternative to convolution in neural networks☆264Mar 28, 2024Updated 2 years ago
- JAX library for training sub-4B foundation models for edge☆306Aug 28, 2024Updated last year
- Reimplementation of `Improving language models by retrieving from trillions of tokens`☆19Nov 16, 2022Updated 3 years ago
- PhD Publications and Thesis on LASSO Model Predictive Control☆20Jun 2, 2019Updated 7 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Multiple dispatch over abstract array types in JAX.☆143Updated this week
- A metrics library for the JAX ecosystem☆41Mar 16, 2023Updated 3 years ago
- ☆11May 1, 2022Updated 4 years ago
- ☆23Jan 23, 2024Updated 2 years ago
- ☆10Aug 18, 2016Updated 10 years ago
- Reproduction of "Scyclone" with PyTorch☆16Jan 6, 2021Updated 5 years ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆218Apr 4, 2021Updated 5 years ago