Ankit Vishwakarma — Senior Software Engineer
Software Engineer focused on GPU compute and low‑level performance optimization. Most of my work revolves around making GPU workloads faster and more efficient, whether that’s through hardware‑aware kernel tuning, system‑level improvements, or getting the most out of modern GPU architectures. I also enjoy contributing to open‑source AI frameworks. I’ve been helping improve PyTorch’s ROCm backend by fixing and enhancing UTs for AMD Instinct GPUs, which helps strengthen reliability, correctness, and overall hardware support for the community. I’m genuinely excited about GPU architecture, kernel performance, and high‑throughput AI systems, and I’m always exploring new ways to squeeze more performance out of the hardware. If you’re working on AI performance engineering, GPU kernel optimization, or any compute‑heavy system, let's connect and learn more about what you’re building.
Stackforce AI infers this person is a GPU compute specialist in the AI and software engineering industry.
Location: Pune, Maharashtra, India
Experience: 7 yrs 7 mos
Skills
- C++
- Rocm
Career Highlights
- Expert in GPU compute and performance optimization.
- Contributed to open-source AI frameworks, enhancing PyTorch's ROCm backend.
- Passionate about high-throughput AI systems and GPU architecture.
Work Experience
AMD
Senior Software Engineer (7 mos)
Freelance
Graphics Programmer (3 yrs 8 mos)
Amdocs
Software Engineer - Advanced (7 yrs)
Education
Bachelor of Engineering at Savitribai Phule Pune University
Higher Secondary Certificate (HSC) at Dr. D. Y. Patil Arts, Scienec & Commerce College, Pimpri, Pune 18.