Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise
☆40Aug 29, 2024Updated 2 years ago
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:
- Implementation of the proposed LVMAE, from the paper, Extending Video Masked Autoencoders to 128 frames, in Pytorch☆55Nov 25, 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
- Implementation of Infini-Transformer in Pytorch☆112Jan 4, 2025Updated last year
- Experimental GPU language with meta-programming☆32Sep 6, 2024Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Fork of HyenaDNA, a long-range genomic foundation model built with Hyena☆10Aug 14, 2023Updated 3 years ago
- some mixture of experts architecture implementations☆28Mar 22, 2024Updated 2 years ago
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆121Oct 17, 2024Updated last year
- Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI☆1,400Updated this week
- PyTorch Implementation of ViT-TTS (EMNLP'23)☆11Oct 20, 2023Updated 2 years ago
- [CVPR 2024] Code and datasets for 'Learning Spatial Features from Audio-Visual Correspondence in Egocentric Videos'☆14Jun 16, 2024Updated 2 years ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆143Updated this week
- ☆13Sep 13, 2025Updated 11 months ago
- PaiNN in jax☆11Jan 14, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Official implementation of the UMDQN algorithm presented in the scientific research paper entitled "Distributional Reinforcement Learning…☆11Jun 3, 2022Updated 4 years ago
- Implementation of CALM from the paper "LLM Augmented LLMs: Expanding Capabilities through Composition", out of Google Deepmind☆177Sep 12, 2024Updated last year
- Implicit Distributional Actor Critic☆12Dec 8, 2021Updated 4 years ago
- The source code of paper "Semantic Enhanced Text-to-SQL Parsing via Iteratively Learning Schema Linking Graph" in KDD2022.☆15Jan 9, 2023Updated 3 years ago
- Utilities for PyTorch distributed☆26Feb 27, 2025Updated last year
- Implementation of a multimodal diffusion transformer in Pytorch☆108Jun 24, 2024Updated 2 years ago
- Implementation of the proposed MaskBit from Bytedance AI☆82Nov 12, 2024Updated last year
- Research code of Cycle Generative Adversarial Networks for Complementary Item Recommendations.☆22Mar 9, 2023Updated 3 years ago
- ☆13Jun 2, 2024Updated 2 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Implementation of the proposed Spline-Based Transformer from Disney Research☆105Nov 9, 2024Updated last year
- # Best Track Data (HURDAT2) Atlantic hurricane database (HURDAT2) 1851-2018 (5.9MB download) This dataset was provided on 10 May 2019 to…☆10Nov 5, 2019Updated 6 years ago
- Implementation of RigidFormer, Learning Rigid Dynamics using Transformers☆32Aug 11, 2026Updated 3 weeks ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆104Dec 22, 2024Updated last year
- ☆13Aug 9, 2022Updated 4 years ago
- Look Ahead Factor models for quantitative investing using deep learning and uncertainty quantification☆10Jun 25, 2020Updated 6 years ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆138Nov 1, 2025Updated 10 months ago
- To mitigate position bias in LLMs, especially in long-context scenarios, we scale only one dimension of LLMs, reducing position bias and …☆12Jun 18, 2024Updated 2 years ago
- Unofficial Implementation of Selective Attention Transformer☆20Oct 31, 2024Updated 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.
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 3 years ago
- ☆21Apr 14, 2026Updated 4 months ago
- [IROS2023]Learning to Solve Tasks with Exploring Prior Behaviours☆13Mar 3, 2024Updated 2 years ago
- Code for paper Rethinking the Data Annotation Process for Multi-view 3D Pose Estimation with Active Learning and Self-Training☆22Apr 18, 2023Updated 3 years ago
- Repository for the code of the paper "Neural Networks Regularization Through Class-wise Invariant Representation Learning".☆12Oct 1, 2017Updated 8 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆101Feb 25, 2023Updated 3 years ago
- [ICLR 2022] "Bayesian Modeling and Uncertainty Quantification for Learning to Optimize: What, Why, and How" by Yuning You, Yue Cao, Tianl…☆14Aug 19, 2022Updated 4 years ago