Pranshu Jain

Lead IOS Developer

Hyderabad, Telangana, India2 yrs 6 mos experience

Most Likely To SwitchAI ML Practitioner

Key Highlights

Expert in building intelligent AI systems.
Proven track record in causal inference and responsible AI.
Active contributor to impactful open-source projects.

Stackforce AI infers this person is a skilled ML Engineer with a focus on AI systems and software development.

Contact

Skills

Core Skills

Machine LearningAi SystemsResponsible AiLarge Language ModelsCausal InferenceSoftware DevelopmentTechnical WritingOpen Source DevelopmentProblem SolvingWeb Development

Other Skills

AI-driven test automationROC ML frameworkskernel validation pipelinesmulti-model evaluationsafety testingLLM evaluationfine-tuninginstruction-tuningCausicaDoWhyAPI DevelopmentReact.jsDockerTechnical ReviewingNode.js

About

[/ML Engineer][begin_profile] I am an ML Engineer focused on building intelligent, real-world AI systems across Agentic AI, NLP, Graph ML, and Causal Inference. Currently, I’m a Co-op Intern at AMD, working on applied AI problems. Previously, I’ve worked as a Research Intern at IIT Madras and a Data Science Fellow at Fellowship.AI, where I built and validated causal graphs using DoWhy, Causica, and Glasso to uncover actionable cause–effect relationships in complex systems. My work spans: • Agentic AI & LLM evaluation • Graph Neural Networks & representation learning • Causal inference for decision-making • Scalable ML systems & experimentation Beyond industry roles, I actively contribute to open-source digital public goods (C4GT, Sunbird ED) and have led multiple hackathon-winning teams focused on impact-driven AI solutions. I’m motivated by problems where AI meets scale, rigor, and societal value — and I enjoy mentoring, collaborating, and shipping systems that actually get used. [end_profile]

Experience

2 yrs 6 mos

Total Experience

9 mos

Average Tenure

11 mos

Current Experience

Amd

Co/Op Intern

Jul 2025 – Present · 10 mos · Hyderabad, Telangana, India · On-site

Building an AI-driven test automation agent, reducing manual QA effort by 30% and improving edge-case defect detection by 10%.
Optimizing and validating ROCm ML frameworks (hipDNN, Flash-Attention), delivering a 4% performance improvement across AMD GPU workloads.
Scaling kernel validation pipelines to ensure correctness and performance across ZEN and CDNA GPU architectures.
Improving cross-hardware compatibility and expanding test coverage for ROCm math libraries and ML frameworks.

AI-driven test automationROC ML frameworkskernel validation pipelinesMachine LearningAI Systems

Indian institute of technology, madras

Research Intern

May 2025 – Jul 2025 · 2 mos · Chennai, Tamil Nadu, India · On-site

Designed a persona-driven Responsible AI safety framework (EduBench), aligned with global AI policy standards, enabling context-aware governance across 9 education-specific risk domains in the Indian ecosystem.
Integrated multi-model evaluation and persona-specific harm perspectives, improving safety assessment robustness by 28% and evaluation accuracy by 20%.
Conducted systematic safety testing across diverse user personas to identify ethical risks and governance gaps in educational AI systems.

Responsible AImulti-model evaluationsafety testing

Outlier

LLM Evaluator

Feb 2025 – May 2025 · 3 mos · Remote

Evaluated LLM responses across diverse tasks, assessing reasoning quality, instruction adherence, domain adaptation, and overall model capability.
Identified fine-tuning and instruction-tuning gaps and curated high-quality supervised fine-tuning (SFT) pairs to improve response quality and consistency.

LLM evaluationfine-tuninginstruction-tuningLarge Language Models

Fellowship.ai

Data Science Fellow

Sep 2024 – Dec 2024 · 3 mos · Kozhikode, Kerala, India · Remote

Applied causal inference techniques using Causica and DoWhy to analyze complex datasets and uncover key cause–effect relationships.
Performed treatment-effect analysis to inform product development decisions, improving decision quality by 25%.
Strengthened causal model reliability through refutation testing, reducing runtime by 12% and improving overall application efficiency by 10%.

causal inferenceCausicaDoWhyCausal Inference

Telaverge communications

Software Engineer Intern

May 2024 – Aug 2024 · 3 mos · Bengaluru, Karnataka, India · Remote

Developed and maintained repositories on Bitbucket and GitHub, ensuring synchronization with project requirements. This initiative led to a 15% increase in productivity and a 25% reduction in project management time.
Designed and implemented custom components such as carousel images and modals, significantly enhancing plugin features and reducing development time by 20%.
Managed two major releases of frontend microservices for the Regal product, ensuring both releases met strict deadlines and quality benchmarks.

API DevelopmentReact.jsDockerSoftware Development

Packt

Technical Reviewer

Feb 2024 – Mar 2024 · 1 mo · Remote

Reviewed ”Node.js for beginners” book, spanning beginner to advanced topics. Enhanced content by offering practical suggestions and real-world examples. Provided in-depth explanations, particularly on complex subjects like error handling and event emitting, ensuring accessibility for all readers.

Technical ReviewingNode.jsJavaScriptTechnical Writing

Code for govtech (c4gt)

Open Source Developer

Jul 2023 – Sep 2023 · 2 mos · Remote

Developed an NPM package for the Sunbird ED-Search widget, reducing dynamic filter development and configuration time by 30% across deployments.
Designed an Algolia InstantSearch–style component with dynamic filters and interactive result rendering, improving search efficiency and user experience.
Contributed to a digital public goods initiative, with the solution deployed on the DIKSHA portal serving 1Cr+ users.

NPM package developmentAlgolia InstantSearchdynamic filtersOpen Source Development

Digilocker

Software Engineer Intern

Dec 2022 – May 2023 · 5 mos · Delhi , India

Worked as the frontend developer on the project ApiSetu a one-place for Api listing and accessing needs for central and state governments.
Ensures that the user experience is seamless and visually appealing

API DevelopmentNext.jsTypeScriptSoftware Development

Parentheses

Software Developer

Aug 2022 – Oct 2022 · 2 mos · Delhi, India

Designed the Admin Dashboard for the project Yaduka Fresh, for regulating various processes of the company.
Developed the front-end interface for project Mamuni, a voice-converter web app that converts one language voice to other regional languages.
Worked as a backend developer for a platform for managing marketing and campaign of organization, Developed synchronous serve and robust updation in web app interlinked with DV360.

API DevelopmentReact.jsFirebaseSoftware Development

Omikron technologies pvt. ltd.

Software Developer Intern

Jun 2022 – Jul 2022 · 1 mo · Remote

Worked with the Redwing team to improve app performance by optimizing API calls and static propping frequently used elements of data.
Received recognition for the Best Feature Addition in the Redwing project (Feature Added: Conditional User States).
Tech-Stack: React.Js, Vanilla-JavaScript, Styled Components

API DevelopmentReact.jsExpress.jsSoftware Development

Girlscript summer of code

Contributor

Apr 2022 – Jun 2022 · 2 mos · Remote

Contributed to 2 projects one of web development and another of competitive programming.
Provide well-written code with taking care of the ethics and legacy of the project.
Added some excellent features in the web development project, such as making the website the material theme, and making it more convenient to present.
Written well explanatory code of some of the advanced data structures like Trie, Tress, and exception handling which makes beginners get them those topics quickly.

Codechef adgitm chapter

2 roles

Problem Tester (Executive Member)

Promoted

Feb 2022 – Jul 2022 · 5 mos

Responsible for setting and testing problems for coding contests
Created and reviewed test cases for different types of coding problems
Conducting different Codechef webinars and Youtube live streams
Educating other students about different concepts of coding

Data Structures and AlgorithmsC/C++Problem Solving

Member

Feb 2021 – Jul 2022 · 1 yr 5 mos

The campus chronicles

Web Developer

Jan 2022 – Jul 2022 · 6 mos · Delhi, India

Girlscript winter of contributing

Contributor

Sep 2021 – Nov 2021 · 2 mos · Remote

Technical Writing Contribution.
Provides well Document content on DSA topics in C/Cpp and Web Optimization as Frontend Developer.
Achieve Position under the top 5 contributors in both domains.

C/C++JavaScriptWeb Development