AlloyMoE: GPU Kernels for Mixture of Experts

In Progress
March 2026 · Arthur Feeney, Ying Wai Li, Aparna Chandramowlishwaran

This project is looking at developing performant and memory efficient GPU kernels for MoE modules.

bern-nn high-level diagram

BERN-NN-IBF: Enhancing Neural Network Bound Propagation Through Implicit Bernstein Form and Optimized Tensor Operations

IEEE TCAD 2024
November 2024 · Wael Fatnassi, Arthur Feeney, Valen Yamamoto, Aparna Chandramowlishwaran, Yasser Shoukry

This paper explores methods to perform efficient bounds-propagation on neural networks.