← Search

Ali Taghibakhshi

6 accepted papers

2026

Scaling Laws and Architectural Frontiers in Metagenomic Foundation Models

ICML 2026poster

Foundation models for genomics have the potential to revolutionize therapeutic design, yet the optimal architectural choices for modeling the vast and diverse distribution of metagenomic data remain under-explored. In this work, we present the machine learning methodology behind MODEL, a family of m…

Cited by 0SourceScholar
2026

Star Elastic: Many-in-One Reasoning LLMs with Efficient Budget Control

ICML 2026poster

Training a family of large language models (LLMs), either from scratch or via iterative compression, is prohibitively expensive and inefficient, requiring separate training runs for each model in the family. In this paper, we introduce Star Elastic, a novel LLM post-training method that adds N neste…

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
2023

MG-GNN: Multigrid Graph Neural Networks for Learning Multilevel Domain Decomposition Methods

ICML 2023poster

Domain decomposition methods (DDMs) are popular solvers for discretized systems of partial differential equations (PDEs), with one-level and multilevel variants. These solvers rely on several algorithmic and mathematical parameters, prescribing overlap, subdomain boundary conditions, and other prope…

2022

Learning Interface Conditions in Domain Decomposition Solvers

NeurIPS 2022accept

Domain decomposition methods are widely used and effective in the approximation of solutions to partial differential equations. Yet the \textit{optimal} construction of these methods requires tedious analysis and is often available only in simplified, structured-grid settings, limiting their use fo…

2021

Optimization-Based Algebraic Multigrid Coarsening Using Reinforcement Learning

NeurIPS 2021poster

Large sparse linear systems of equations are ubiquitous in science and engineering, such as those arising from discretizations of partial differential equations. Algebraic multigrid (AMG) methods are one of the most common methods of solving such linear systems, with an extensive body of underlying…