← Search

Ameya Sunil Mahabaleshwarkar

4 accepted papers

2026

Orak: A Foundational Benchmark for Training and Evaluating LLM Agents on Diverse Video Games

ICLR 2026poster

Large Language Model (LLM) agents are reshaping the game industry, by enabling more intelligent and human-preferable characters. Yet, current game benchmarks fall short of practical needs: they lack evaluations of diverse LLM capabilities across various game genres, studies of agentic modules crucia…

Cited by 0SourceScholar
2025

Efficient Hybrid Language Model Compression through Group-Aware SSM Pruning

NeurIPS 2025poster

Hybrid language models that combine Attention and State Space Models (SSMs) have been shown to achieve state-of-the-art accuracy and runtime performance. Recent work has also demonstrated that applying pruning and distillation to Attention-only models yields smaller, more accurate models at a fracti…

Cited by 0SourceScholar
2025

Hymba: A Hybrid-head Architecture for Small Language Models

ICLR 2025spotlight

We propose Hymba, a family of small language models featuring a hybrid-head parallel architecture that integrates attention mechanisms and state space models (SSMs) within the same layer, offering parallel and complementary processing of the same inputs. In this hybrid-head module, attention heads p…

Cited by 12SourcePDFScholar