PyTorch interface for TrueGrad Optimizers
☆43Aug 8, 2023Updated 2 years ago
Alternatives and similar repositories for TrueGrad
Users that are interested in TrueGrad are comparing it to the libraries listed below
Sorting:
- Automatically take good care of your preemptible TPUs☆37May 15, 2023Updated 2 years ago
- ☆18Aug 24, 2024Updated last year
- ☆33Nov 4, 2024Updated last year
- Focused on fast experimentation and simplicity☆80Dec 24, 2024Updated last year
- High performance pytorch modules☆17Jan 14, 2023Updated 3 years ago
- ☆13Jan 15, 2025Updated last year
- Parallel Associative Scan for Language Models☆18Jan 8, 2024Updated 2 years ago
- ARC Community Project☆22Aug 2, 2024Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Jan 20, 2024Updated 2 years ago
- Doing style transfer with linguistic features using OpenAI's CLIP.☆14May 4, 2021Updated 4 years ago
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Jun 21, 2023Updated 2 years ago
- research impl of Native Sparse Attention (2502.11089)☆63Feb 19, 2025Updated last year
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆23Oct 15, 2024Updated last year
- ☆23Jun 18, 2024Updated last year
- This is an implementation of Image2StyleGAN embedding algorithm and various experiments using StyleGAN2-ADA as backbone.☆17Sep 2, 2021Updated 4 years ago
- This tool displays tflite signatures and rewrites the input/output OP name to the name of the signature. There is no need to install Tens…☆14Dec 13, 2023Updated 2 years ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆18Jul 24, 2025Updated 7 months ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- LoRA for arbitrary JAX models and functions☆145Feb 26, 2024Updated 2 years ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆13Jul 19, 2024Updated last year
- A simple PyTorch implementation of CLIP model using DinoV2 and BERT☆15Sep 26, 2023Updated 2 years ago
- ☆13Nov 27, 2025Updated 3 months ago
- minGPT in JAX☆48Jan 10, 2022Updated 4 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Aug 6, 2021Updated 4 years ago
- ☆69Mar 24, 2025Updated 11 months ago
- ☆10Apr 5, 2024Updated last year
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- Awesome Triton Resources☆39Apr 27, 2025Updated 10 months ago
- Bring ROS to any Linux distributions.☆15Jun 12, 2019Updated 6 years ago
- Efficient optimizers☆294Updated this week
- Official Repository for ICML 2023 paper "Can Neural Network Memorization Be Localized?"☆21Oct 26, 2023Updated 2 years ago
- Image2StyleGAN and Image2StyleGAN++ implementation☆28Jul 15, 2021Updated 4 years ago
- WIP☆94Aug 13, 2024Updated last year
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- A toolkit for scaling law research ⚖☆58Jan 27, 2025Updated last year
- ☆34Sep 10, 2024Updated last year
- ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.☆55Sep 18, 2022Updated 3 years ago
- ☆255Dec 2, 2024Updated last year
- Brax + Pufferlib + CARBS for gpu-accelerated robotics RL☆12Jun 12, 2025Updated 9 months ago