Data Scientist

1 day ago


Vellore, India FPX AI Full time
Role Overview

FPX is building an AI infrastructure marketplace that enables developers to access and deploy compute efficiently. As a PyTorch + CUDA Engineer focused on benchmarking and performance, you will be responsible for designing, running, and interpreting benchmarks across model, framework, and hardware stacks. Your role is critical in validating performance claims, detecting regressions, and guiding optimizations to ensure FPX remains at the cutting edge of compute efficiency.

You will collaborate closely with ML systems, compiler, hardware, and platform teams. You should have strong experience in PyTorch internals, GPU programming, and profiling tools.


Key Responsibilities
  • Define, build, and maintain a benchmark suite covering representative deep learning workloads (training, inference, mixed) across modalities (vision, NLP, recommendation, etc.).
  • Automate running of benchmarks across multiple hardware configurations (NVIDIA GPUs, possibly AMD, and future accelerators).
  • Use profiling, tracing, and performance tools (e.g. Nsight Systems, Nsight Compute, PyTorch Profiler, CUPTI, NVTX) to identify bottlenecks across layers (operator, kernel, memory, data movement).
  • Write and maintain scripts / harnesses that manage benchmark orchestration, result collection, and analysis (latency, throughput, memory usage, utilization metrics).
  • Detect and triage performance regressions (e.g. nightly, CI-integrated benchmarks).
  • Partner with compiler / runtime / kernel teams to propose optimizations, micro-bench kernel patches, fusion, operator-level improvements, or configuration tuning.
  • Validate performance improvements across scale (multi-GPU, distributed) and in production-like settings.
  • Publish benchmark results, document methodology, and communicate trade-offs to stakeholders (engineering, product, customers).
  • Occasionally assist in custom kernel development when needed (e.g. fused kernels, optimized CUDA code) or integrating specialized libraries (Triton, CUTLASS, cuBLAS, cuDNN).
  • Stay up-to-date on new features in PyTorch (e.g. torch.compile, CUDA Graphs, new backends) and evaluate their impact.


Required Qualifications
  • BS / MS / PhD in Computer Science, Electrical Engineering, or equivalent experience.
  • Solid experience (3+ years) in GPU programming: CUDA, kernel development, memory management, concurrency.
  • Deep familiarity with PyTorch internals (operators, autograd, dispatcher, JIT/inductor pipeline or equivalent).
  • Experience with profiling and analysis of GPU workloads (Nsight, CUPTI, NVTX, PyTorch Profiler).
  • Strong Python and C++ skills.
  • Ability to analyze low-level performance (latency, throughput, memory, occupancy) and correlate to high-level model behavior.
  • Experience writing benchmark harnesses, automation, and result pipelines.
  • Excellent communication skills — able to present performance trade-offs and complex analysis to technical and non-technical audiences.


Preferred / Nice-to-Have
  • Experience with distributed training/inference (DDP, FSDP, model parallelism).
  • Experience with PyTorch’s newer compilation pathways (e.g. torch.compile, Inductor, Dynamo).
  • Knowledge of CUDA Graphs, kernel fusion, memory optimizations, tensor core usage.
  • Experience with other ML frameworks and baselining comparisons (TensorFlow, JAX, ONNX).
  • Published benchmarks, open-source contributions, or performance tools development.
  • Prior experience in systems, compilers, or GPU runtime development.
  • Familiarity with scaling benchmarks, cluster deployments, and heterogeneous hardware.


Compensation
  • Competitive salary + equity + benefits.
  • Potential for bonuses tied to performance improvements and critical benchmark delivery.

  • Data scientist

    2 weeks ago


    Vellore, India InvokHR Solutions Full time

    We are hiring - Data Scientist!About Client:We are an innovative, stealth-mode startup at the forefront of applying next-generation AI to chemistry and scientific reasoning. As a sister concern of Laxai Life Sciences and working closely with our clients, we are uniquely positioned to revolutionize how groundbreaking molecules are discovered and developed,...

  • Data Scientist

    3 weeks ago


    Vellore, Tamil Nadu, India Opvia Full time

    Location: HybridExperience Level: 1–3 yearsWorking Hours: UK timings (flexibility may be discussed)About the RoleWe are seeking a Data Scientist to join our team and drive innovation through data. This role is focused on building and deploying ML models, enabling product recommendations, and unlocking customer insights that directly impact business growth....

  • Data Scientist

    6 days ago


    Vellore, India STRYDO TECHNOLOGIES PVT.LTD Full time

    We are looking for an expert in machine learning to help us extract value from our data. You will lead all the processes from data collection, cleaning, and pre-processing, to training models and deploying them to production. Schedule: - Day shift

  • Data Scientist

    6 days ago


    Vellore, India STRYDO TECHNOLOGIES PVT.LTD Full time

    We are looking for an expert in machine learning to help us extract value from our data. You will lead all the processes from data collection, cleaning, and pre-processing, to training models and deploying them to production. **Salary**: ₹15,000.00 per month

  • Lead Data Scientist

    4 weeks ago


    Vellore, Tamil Nadu, India Housing Full time

    Who we are...?REA India is a part of REA Group Ltd. of Australia (ASX: REA) ("REA Group"). It is the country's leading full stack real estate technology platform that owns Housing.com.In December 2020, REA Group acquired a controlling stake in REA India. REA Group, headquartered in Melbourne, Australia, is a multinational digital advertising business...

  • Data Modeler

    2 weeks ago


    Vellore, Tamil Nadu, India beBeeMachineLearning Full time ₹ 25,00,000 - ₹ 40,00,000

    Unlock a career in predictive analytics by joining our team as a Data Scientist.Key Responsibilities:Develop, train, and validate machine learning models using various techniques.Collaborate with cross-functional teams to define data requirements and success metrics.Deploy machine learning models into production using industry best practices.Build automated...

  • Senior data engineer

    3 weeks ago


    Vellore, Tamil Nadu, India NovintiX Full time

    Novinti X is seeking a highly skilled and experienced Data Engineers (x4) to join our growing data team focused on delivering impactful solutions in the healthcare and medical technology space. If you're passionate about building scalable pipelines, driving analytics, and working with healthcare data standards—this role is for you???? ????????????

  • Data Steward

    4 weeks ago


    Vellore, Tamil Nadu, India InvokHR Solutions Full time

    We are hiring for Data Steward LeadAbout the client A boutique AI and data engineering firm, founded in 2018 and headquartered in Gurgaon, India, that empowers Fortune 1000 companies through AI-powered decision intelligence—especially in finance and supply chain transformation—delivering over US $2 billion in financial benefits while fostering a highly...


  • Vellore, India Ampstek Full time

    Title: Senior Generative AI Engineer (Databricks Data Lake)Location: Remote (India)Full Time Job Summary:About the RoleWe are seeking an experienced Senior Generative AI Engineer with a strong background in Databricks and data lake architectures. This individual will be responsible for designing, developing, and deploying cutting-edge Generative AI (GenAI)...


  • Vellore, Tamil Nadu, India Mitchell Martin Inc. Full time

    Position OverviewThe Engineer, Artificial Intelligence supports the development and deployment of advanced artificial intelligence solutions. This role focuses on implementing solutions that leverage Machine Learning, Natural Language Processing, and emerging technologies such as Generative AI and Large Language Models. The Engineer, Artificial Intelligence...