A repo based on XiLin Li's PSGD repo that extends some of the experiments.
☆14Oct 7, 2024Updated last year
Alternatives and similar repositories for Preconditioned-Stochastic-Gradient-Descent
Users that are interested in Preconditioned-Stochastic-Gradient-Descent are comparing it to the libraries listed below
Sorting:
- ☆18Aug 24, 2024Updated last year
- DiT (training + flow matching) in Jax☆11Jan 5, 2025Updated last year
- Tensorflow implementation of MuZero algorithm☆11Aug 23, 2022Updated 3 years ago
- Official PyTorch implementation of "EdVAE: Mitigating Codebook Collapse with Evidential Discrete Variational Autoencoders"☆14Sep 20, 2024Updated last year
- ☆21Jan 23, 2024Updated 2 years ago
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated last year
- Automatically take good care of your preemptible TPUs☆37May 15, 2023Updated 2 years ago
- Using a modified version of Werner Duvaud's MuZero implementation (https://github.com/werner-duvaud/muzero-general) this reinforcement ag…☆18Jun 30, 2021Updated 4 years ago
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆43Sep 19, 2022Updated 3 years ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Awesome Medical Imaging Datasets☆48Dec 1, 2025Updated 3 months ago
- Resilient Model-Based RL by Regularizing Posterior Predictability☆22Mar 4, 2024Updated 2 years ago
- PyTorch implementation of BPDA+EOT attack to evaluate adversarial defense with an EBM☆26Jun 30, 2020Updated 5 years ago
- Official repository for paper "Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matching" (ICML…☆28Jan 12, 2023Updated 3 years ago
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆30Jul 18, 2023Updated 2 years ago
- Large-scale remote sensing image target recognition and automatic annotation☆12Nov 23, 2024Updated last year
- Dice Language Support for VS Code☆10Sep 29, 2020Updated 5 years ago
- A minimal Pytorch Implementation of Stochastically Quantized Variational AutoEncoder (SQ-VAE) by Sony☆33Oct 16, 2023Updated 2 years ago
- Implementation of PSGD optimizer in JAX☆35Dec 31, 2024Updated last year
- Implementation of MuZero with PyTorch, based on the pseudocode from DeepMind (https://arxiv.org/src/1911.08265v2/anc/pseudocode.py).☆33Aug 14, 2022Updated 3 years ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆40Feb 9, 2026Updated 3 weeks ago
- DiMSUM: Diffusion Mamba - A Scalable and Unified Spatial-Frequency Method for Image Generation (NeurIPS 2024)☆43Feb 18, 2025Updated last year
- ☆10Apr 5, 2024Updated last year
- ☆17Feb 4, 2026Updated last month
- Russian words synonyms and antonyms☆11Dec 7, 2021Updated 4 years ago
- EOSIO-Taurus - The Most Powerful Infrastructure for Decentralized Applications☆13Mar 29, 2024Updated last year
- ☆14Feb 25, 2026Updated last week
- Paper survey of efficient computation for large scale models.☆34Dec 7, 2024Updated last year
- Fast singularity detection with kernel☆38Jan 4, 2024Updated 2 years ago
- ☆234Feb 12, 2025Updated last year
- ☆35Dec 5, 2022Updated 3 years ago
- ☆10Apr 16, 2023Updated 2 years ago
- RS-IMLE☆44Dec 7, 2024Updated last year
- Python package providing functionality and plotting for chemistry method comparison☆16Feb 28, 2024Updated 2 years ago
- VisionGRU: A Linear-Complexity RNN Model for Efficient Image Analysis☆13Dec 26, 2024Updated last year
- Advantage Alignment Algorithms (ICLR 2025 oral)☆17Apr 7, 2025Updated 10 months ago
- Not just a PDE toolbox. Adapt your ideas from a clean, modular code base with Femeko.☆15Feb 22, 2026Updated last week
- Firefox and Chrome compatible extension that acts as annotation tool for websites (Named Entity Recognition)☆10Feb 17, 2019Updated 7 years ago
- Official code for PLoP☆17Jun 30, 2025Updated 8 months ago