← Search

Mihai Capotă

3 accepted papers

2025

AutoParLLM: GNN-guided Context Generation for Zero-Shot Code Parallelization using LLMs

NAACL 2025long

In-Context Learning (ICL) has been shown to be a powerful technique to augment the capabilities of LLMs for a diverse range of tasks. This work proposes AutoParLLM, a novel way to generate context using guidance from graph neural networks (GNNs) to generate efficient parallel codes. We evaluate Auto…

2024

A Structure-Aware Framework for Learning Device Placements on Computation Graphs

NeurIPS 2024poster

Computation graphs are Directed Acyclic Graphs (DAGs) where the nodes correspond to mathematical operations and are used widely as abstractions in optimizations of neural networks. The device placement problem aims to identify optimal allocations of those nodes to a set of (potentially heterogeneous…

2020

Approximating Stacked and Bidirectional Recurrent Architectures with the Delayed Recurrent Neural Network

ICML 2020poster

Recent work has shown that topological enhancements to recurrent neural networks (RNNs) can increase their expressiveness and representational capacity. Two popular enhancements are stacked RNNs, which increases the capacity for learning non-linear functions, and bidirectional processing, which expl…