About
AI Engineer | 5 years building production AI systems | Three published research papers | M.S. in Electrical Engineering from IIT Madras I build intelligent systems that don't just look good on paper, —they scale, optimize, and deliver. At SuperAGI, I designed AI agent frameworks that automate complex workflows while keeping costs lean (recently cut LLM expenses by 12% without sacrificing performance). My background spans the full AI stack—from low-level GPU optimization with NVIDIA's DGX systems to high-level LLM agent orchestration. I've worked on medical AI with GE HealthCare (transforming cancer treatment guidelines into queryable data), benchmarked supercomputers at IIT Madras's Brain Centre, and even built communication systems for a Hyperloop pod that competed in the Netherlands. What gets me excited: Hard problems where vanilla solutions don't cut it. Model compression, quantization, knowledge distillation, CUDA-level optimizations—I love diving deep into what makes AI systems actually perform in production. Core competencies: Deep Learning & Computer Vision (PyTorch, TensorFlow, CNNs, GANs, Transformers) LLM integration & AI agent systems (LangChain, prompt engineering) GPU programming & optimization (CUDA, TensorRT, ONNX, model profiling) Backend engineering & MLOps (FastAPI, Docker, Git) Data processing & analysis (NumPy, Pandas, Scikit-learn) Programming: Python, C++, C, Java Research Papers: FastDTI: 3D Super-Resolution for DTI (SASHIMI 2025) - [https://doi.org/10.1007/978-3-032-05573-6_13] Knowledge Models for Cancer Guidelines (IEEE EMBC 2024) - [https://pubmed.ncbi.nlm.nih.gov/40039144] Automated Knowledge Modeling (IEEE EMBC 2023) - [https://pubmed.ncbi.nlm.nih.gov/38082992] Always open to connecting with engineers and researchers pushing the boundaries of what AI can do. Whether you're optimizing models, building agent systems, or working on something ambitious—let's talk! 💬 GitHub: github.com/arihantjain124
Work experience
Member of Technical Staff, AI Engineering at Terrabase, Nov 2025 - Present · 10 mos
AI Engineer at SuperAGI, Aug 2024 - Oct 2025 · 1 yr 3 mos, Bengaluru, Karnataka, India, - SuperAGI is a CRM platform that uses AI Agents to automate and optimize the sales pipeline. I work in the research team, where we rapidly prototype various frameworks and libraries to build robust and intelligent AI agents. - Experimenting with the latest development in LLMs by building AI Agent for task-specific application, this involves connecting LLMs with other services to run mundane repeatable task without much human intervention.
Research Associate at Sudha Gopalakrishnan Brain Center, Nov 2022 - Jul 2024 · 1 yr 9 mos, Chennai, Tamil Nadu, India, ● Worked jointly with Engineers at NVIDIA on benchmarking the DGX supercomputer system at the Computational Group of Sudha Gopalakrishnan Brain Centre for GPU profiling and model precision variation to accelerate extensive data deep learning-based workloads across multiple GPUs and multiple compute nodes
Research Associate at IITMs HTIC MedTech Incubator, Sep 2021 - Aug 2023 · 2 yrs, Chennai, Tamil Nadu, India, ● Worked in collaboration with GE HealthCare on extracting clinical cancer guidelines from NCCNs pdf to make the data queryable and computable. ● The data extracted allows patient reports to map to specific points in the cancer treatment, making it easier to follow procedure guidelines
Embedded Subsystem Engineer at Team Avishkar Hyperloop, Sep 2021 - Aug 2022 · 1 yr, Chennai, Tamil Nadu, India, ● Worked extensively on the communication platform for the hyperloop pod. ● Tested various failure scenarios with the control code to proof test the pod. ● Helped in the integration of various subsystems with embedded hardware control.
Education
Master of Science - MS, Electrical and Electronics Engineering, Indian Institute of Technology, Madras
Bachelor of Engineering, Computer Science, Walchand Institute of Technology, Solapur
