Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise
☆40Aug 29, 2024Updated last year
Alternatives and similar repositories for sinkhorn-router-pytorch
Users that are interested in sinkhorn-router-pytorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Code for the paper: Comparing and Contrasting Deep Learning Weather Prediction Backbones on Navier-Stokes and Atmospheric Dynamics☆15Aug 9, 2024Updated last year
- 삼각형의 실전! Triton☆16Feb 15, 2024Updated 2 years ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Sep 23, 2024Updated last year
- Experimental GPU language with meta-programming☆27Sep 6, 2024Updated last year
- Implementation of Infini-Transformer in Pytorch☆112Jan 4, 2025Updated last year
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- PyTorch Implementation of ViT-TTS (EMNLP'23)☆11Oct 20, 2023Updated 2 years ago
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆1,328Jan 27, 2026Updated 2 months ago
- Pytorch implementation of various token mixers; Attention Mechanisms, MLP, and etc for understanding computer vision papers and other tas…☆16Mar 11, 2026Updated 2 weeks ago
- ☆21Mar 3, 2025Updated last year
- ☆12Sep 13, 2025Updated 6 months ago
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- ☆27Mar 29, 2025Updated last year
- Implementation of a multimodal diffusion transformer in Pytorch☆107Jun 24, 2024Updated last year
- Implicit Distributional Actor Critic☆11Dec 8, 2021Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- ☆14Mar 8, 2025Updated last year
- Official code release for Deep Extreme Mixture Model by Wilson, McDonald, Galib, Tan, and Luo.☆10Feb 11, 2022Updated 4 years ago
- Implementation of the proposed MaskBit from Bytedance AI☆82Nov 12, 2024Updated last year
- Utilities for PyTorch distributed☆25Feb 27, 2025Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆103Dec 22, 2024Updated last year
- Code for Novel View Acoustic Synthesis paper☆51Aug 14, 2023Updated 2 years ago
- Benchmarking scripts for Gaia☆14Apr 10, 2025Updated 11 months ago
- ☆13Jun 2, 2024Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Wordpress hosting with auto-scaling on Cloudways • AdFully Managed hosting built for WordPress-powered businesses that need reliable, auto-scalable hosting. Cloudways SafeUpdates now available.
- ☆52Aug 19, 2024Updated last year
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 2 years ago
- Unofficial Implementation of Selective Attention Transformer☆21Oct 31, 2024Updated last year
- Code for paper Rethinking the Data Annotation Process for Multi-view 3D Pose Estimation with Active Learning and Self-Training☆22Apr 18, 2023Updated 2 years ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆135Nov 1, 2025Updated 4 months ago
- ☆14Apr 26, 2018Updated 7 years ago
- Inverse Scaling in Test-Time Compute☆25Dec 3, 2025Updated 3 months ago
- To mitigate position bias in LLMs, especially in long-context scenarios, we scale only one dimension of LLMs, reducing position bias and …☆11Jun 18, 2024Updated last year
- Implementation of a holodeck, written in Pytorch☆18Nov 1, 2023Updated 2 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- [IROS2023]Learning to Solve Tasks with Exploring Prior Behaviours☆12Mar 3, 2024Updated 2 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆102Feb 25, 2023Updated 3 years ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆126Jul 26, 2024Updated last year
- ☆10Feb 10, 2023Updated 3 years ago
- OLD REPOSITORY, new one at repo.rumpkernel.org/rumprun☆44Apr 13, 2015Updated 10 years ago
- ☆12Oct 10, 2023Updated 2 years ago
- ☆13Aug 9, 2022Updated 3 years ago