Udbhav Bamba

ubamba98@gmail.com

I am an Applied Scientist II at Amazon, where I work at the intersection of large language models and low-latency on-device inference. Beyond my industry role, I actively mentor students and collaborate with researchers. Prior to Amazon, I focused on resource-efficient machine learning as an intern at Mila - Quebec AI Institute and co-founded Transmute AI Labs to mentor underrepresented students at IIT Dhanbad.

These days I am working on building better mixture-of-experts systems and helping LLMs reason efficiently under strict compute budgets. Beyond academic research, I am an active competitor on Kaggle, where I hold the Competitions Master title with five gold and eight silver medals. Apart from work, I enjoy gaming, binge-watching shows, and playing badminton.

For updated details, please see my Google Scholar / LinkedIn pages.

Selected publications

Partial Binarization of Neural Networks for Budget-Aware Efficient Learning

Udbhav Bamba, Neeraj Anand, Saksham Aggarwal, and 2 more authors

In IEEE/CVF Winter Conference on Applications of Computer Vision (WACV), 2024

Bib PDF

@inproceedings{Bamba_2024_WACV,
  author = {Bamba, Udbhav and Anand, Neeraj and Aggarwal, Saksham and Prasad, Dilip K. and Gupta, Deepak K.},
  title = {Partial Binarization of Neural Networks for Budget-Aware Efficient Learning},
  booktitle = {IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)},
  year = {2024},
  url = {https://ieeexplore.ieee.org/document/10483283},
}

ChipNet: Budget-Aware Pruning with Heaviside Continuous Approximations

Udbhav Bamba, Rishabh Tiwari, Arnav Chavan, and 1 more author

In International Conference on Learning Representations (ICLR), 2021

Bib PDF

@inproceedings{tiwari2021chipnet,
  title = {ChipNet: Budget-Aware Pruning with Heaviside Continuous Approximations},
  author = {Bamba, Udbhav and Tiwari, Rishabh and Chavan, Arnav and Gupta, Deepak K.},
  booktitle = {International Conference on Learning Representations (ICLR)},
  year = {2021},
  url = {https://openreview.net/forum?id=xCxXwTzx4L1},
}

An UltraMNIST classification benchmark to train CNNs for very large images

Udbhav Bamba, Deepak K. Gupta, Abhishek Thakur, and 6 more authors

In Scientific Data, 2024

Bib PDF

@inproceedings{Gupta2024UltraMNIST,
  title = {An UltraMNIST classification benchmark to train CNNs for very large images},
  author = {Bamba, Udbhav and Gupta, Deepak K. and Thakur, Abhishek and Gupta, Akash and Agarwal, Rohit and Sharan, Suraj and Demir, Ertugul and Agarwal, Krishna and Prasad, Dilip K.},
  booktitle = {Scientific Data},
  year = {2024},
  url = {https://doi.org/10.1038/s41597-024-03587-4},
}

Dynamic Kernel Selection for Improved Generalization and Memory Efficiency in Meta-learning

Arnav Chavan, Rishabh Tiwari, Udbhav Bamba, and 1 more author

In IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2022

Bib PDF

@inproceedings{Chavan2021MetaDock,
  author = {Chavan, Arnav and Tiwari, Rishabh and Bamba, Udbhav and Gupta, Deepak K.},
  booktitle = {IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
  title = {Dynamic Kernel Selection for Improved Generalization and Memory Efficiency in Meta-learning},
  year = {2022},
  url = {https://ieeexplore.ieee.org/document/9878632},
}