Prateek Chokse

Senior Software Engineer

Bengaluru, Karnataka, India7 yrs 8 mos experience

AI ML PractitionerHighly Stable

Key Highlights

Over 6 years of experience in AI and System Design.
Expert in optimizing AI/ML algorithms for performance.
Proven track record in developing scalable solutions.

Stackforce AI infers this person is a skilled AI/ML engineer with a focus on system design and optimization.

Contact

Skills

Core Skills

Ai/ml OptimizationSystem Design

Other Skills

Parallel ProgrammingComputer VisionC++CUDACUDNNTensorRTPythonImage ProcessingGitHubDeep LearningPyTorchMachine LearningLarge Language Models (LLM)Object-Oriented Programming (OOP)Multithreading

About

With over 6 years of experience in the dynamic field of AI, GPU Programming, System Design, and Cloud Computing, I bring a proven record of delivering efficient and scalable solutions. My expertise lies in architecting robust systems that cater to intensive computational needs while optimizing for performance and cost. An adaptive person with an innovative mindset, I excel at adapting to recent technologies and methodologies, ensuring that the systems I design are state-of-the-art and future-proof. My passion for technology drives me to continuously explore innovative advancements, making me an asset to any forward-thinking organization.

Experience

7 yrs 8 mos

Total Experience

6 yrs 11 mos

Average Tenure

9 mos

Current Experience

Amd

Senior Software System Designer

Aug 2025 – Present · 9 mos · Bengaluru, Karnataka, India · Hybrid

Aira matrix private limited

2 roles

Senior Software Engineer

Promoted

Jun 2022 – Aug 2025 · 3 yrs 2 mos · Mumbai Area, India

Interacting with clients and understanding their needs and developing solutions for those needs.
Design and development of Analysis Platform, which is a multi-tier scalable platform for submitting task of various kinds of analysis workload. It has ability to run and manage long running tasks in a GPU cluster.
Optimization of AI/ML algorithms, which usually includes porting of Python code to C++, writing GPU kernels, optimization of data pipelines and multi-threading.
Development and maintenance of CUDNN-based Analysis framework, which includes supporting various new network layers and optimising existing layers.
Integrated the OnnxRuntime and TensorRT with the existing Analysis framework to support the '.onnx' format and added support to custom layers.
Deployment of the Analysis platform in closed and open environments. Having different kind of clusters like Kubernetes, slurm, AWS ECS, etc.
Optimization of Image processing algorithms using GPU and CPU parallel computing and created python bindings for these algorithms for other team usage.
Created internal tools for optimization of the models and facilitate of porting.
Ported openslide library in C++ and added support for other formats like VSI CellSens(.vsi), BigTiff, etc.

Parallel ProgrammingComputer VisionAI/ML OptimizationSystem Design