← Search

Divya Mahajan

4 accepted papers

2025

CATransformers: Carbon Aware Transformers Through Joint Model-Hardware Optimization

NeurIPS 2025poster

Machine learning solutions are rapidly adopted to enable a variety of key use cases, from conversational AI assistants to scientific discovery. As the adoption of machine learning models becomes increasingly prevalent, the associated lifecycle carbon footprint is expected to increase, including both…

Cited by 0SourcecodeScholar
2024

Integrated Hardware Architecture and Device Placement Search

ICML 2024poster

Distributed execution of deep learning training involves a dynamic interplay between hardware accelerator architecture and device placement strategy. This is the first work to explore the co-optimization of determining the optimal architecture and device placement strategy through novel algorithms,…

2023

FLuID: Mitigating Stragglers in Federated Learning using Invariant Dropout

NeurIPS 2023poster

Federated Learning (FL) allows machine learning models to train locally on individual mobile devices, synchronizing model updates via a shared server. This approach safeguards user privacy; however, it also generates a heterogeneous training environment due to the varying performance capabilities ac…

2020

Efficient Algorithms for Device Placement of DNN Graph Operators

NeurIPS 2020poster

Modern machine learning workloads use large models, with complex structures, that are very expensive to execute. The devices that execute complex models are becoming increasingly heterogeneous as we see a flourishing of Domain Specific Architectures (DSAs) being offered as hardware accelerators in a…