45 citations · 50 across the 8 of their papers we have counts for
Showing cs.LGShow all
2 papers · 1 filter
cs.LG2025
Characterizing the Behavior of Training Mamba-based State Space Models on GPUs
Trinayan Baruah, Kaustubh Shivdikar, Sara Prescott +1
Mamba-based State Space Models (SSM) have emerged as a promising alternative to the ubiquitous transformers. Despite the expressive power of transformers, the quadratic complexity…
cs.LG2023★ 3 cited
MaxK-GNN: Extremely Fast GPU Kernel Design for Accelerating Graph Neural Networks Training
Hongwu Peng, Xi Xie, Kaustubh Shivdikar +6
In the acceleration of deep neural network training, the GPU has become the mainstream platform. GPUs face substantial challenges on GNNs, such as workload imbalance and memory acc…