Role: Pytorch Engineer
Location : Bangalore, Hyderabad , Pune
Years of Experience : 5-8 Years
Notice Period : Immediate,15 Days
Mandatory skills : Python ,Pytorch , C++
- Python & C/C++: Expert systems-level programming and tooling development.
- PyTorch & Triton: Hands-on experience developing custom GPU kernels using PyTorch and Triton.
- GEMM & GPU Tuning: Direct experience writing GEMM kernels and optimizing hardware execution.
- Model Porting: Proven ability to port and optimize AI/ML models on custom SDKs and compilers.
- HPC Benchmarking: Hands-on execution of stress tests and benchmarks (e.g., HPL, BERT).
- Compiler & Systems: Deep knowledge of GPU architecture, runtime execution, and compiler optimizations.
If interested, please reach out to
[email protected]
📌 Pytorch Engineer (Bengaluru)
🏢 CIEL HR
📍 Bengaluru