Roshani Narasimhan
Software Engineer, AI/ML @ Google
- Role
- Software Engineer at Google
- Location
- Seattle, WA, US
- LinkedIn followers
- 500 followers
About Roshani Narasimhan
I’m enthusiastic about modeling real life problems into computer science problems, to find meaningful insights. I’m interested in various domains - machine learning at scale, network science, computational biology, and software engineering. I value empathy and leadership, and try to bring the same into my work everyday!
Experience
Software Engineer
Aug 2022 — Present · Seattle, WA, US
Scaling ML models on Cloud TPUs- Pathways on Cloud - distributed ML training runtime, OSS implementation in XPK (GKE job orchestration tool)- Pathways on Vertex AI workbench for an easier JAX development experience- Improved data loading and checkpointing speeds by 5x, in collaboration with GCS. Identified bottlenecks in the ML stack using Maxtext (JAX based LLM)- Developer of PathwaysJob, GKE API for orchestration of Pathways-JAX workloads.
Education
Dhirubhai Ambani University
Bachelor of Technology - BTech, Information and Communication Technology, Minors in Computational Science
North Carolina State University
Master of Science - MS, Computer Science
Find verified contacts for anyone on LinkedIn
Unifers gives sales teams verified emails and direct dials, enriched profiles, and outreach that lands in the inbox.
Free plan included · No credit card required
This profile is compiled from publicly available professional sources. Unifers is not affiliated with or endorsed by LinkedIn. Request removal of this profile.