Vacancy description
Recruitment Room
$145600 USD - $187200 USD / hour
India
Trainium (NKI) Kernel Expert in work from home, India is listed on Jobeax. Browse 30,000+ vacancies available.
Remote | Independent Contractor | United States | $145,600–$187,200 annualized ($70–$90/hour)
Apply your specialized AWS Trainium and Neuron Kernel Interface (NKI) expertise to help improve the quality and reliability of advanced AI systems.
As a Trainium (NKI) Kernel Expert, you’ll evaluate NKI development tasks used to train and assess frontier AI models. You’ll assess whether kernels are technically correct, appropriately designed for Trainium hardware, and optimized for the platform.
Your expertise will be particularly valuable when evaluating CUDA-to-NKI migrations, Trainium-specific performance optimization, and cross-platform numerical correctness .
Evaluate NKI kernel development tasks for quality, correctness, completeness, and hardware appropriateness .
Assess the fidelity and technical quality of CUDA-to-NKI migrations .
Evaluate Trainium-specific kernel optimization strategies and performance considerations.
Review tile-based computation approaches and NKI memory-management patterns.
Assess the appropriate use of SBUF, PSUM, and HBM within kernel implementations.
Evaluate partition-dimension constraints and DMA orchestration.
Assess Trainium kernel performance, including:
NeuronCore pipeline utilization
Tensor-engine throughput
Memory-bandwidth bottlenecks
Evaluate cross-platform numerical correctness between GPU and Trainium implementations.
Assess differences arising from accumulation order, rounding behavior, and mixed-precision semantics.
Provide clear, structured, rubric-based written feedback on kernel tasks and solutions.
2+ years of hands-on experience developing or optimizing kernels with Neuron Kernel Interface (NKI) for AWS Trainium or Inferentia2 hardware.
Strong understanding of NKI development patterns, including:
Tile-based computation
SBUF, PSUM, and HBM memory hierarchy
Partition-dimension constraints
DMA orchestration
Demonstrated experience assessing or performing CUDA-to-NKI kernel migrations .
Familiarity with Trainium-specific performance profiling and optimization.
Strong understanding of cross-platform numerical correctness, including:
Accumulation order
Rounding behavior
Mixed-precision semantics
Ability to analyze technical implementations with precision and communicate findings clearly in writing.
Experience in the following areas is highly valuable:
Hands-on experience with the AWS Neuron SDK .
Familiarity with Neuron Compiler internals.
Contributions to NKI kernel libraries or related projects.
Previous CUDA or Triton kernel development experience.
Strong understanding of Trainium hardware architecture and capabilities, including:
NeuronCore-v2 architecture
On-chip SRAM topology
FP32, BF16, FP8, and INT8 data types
Experience benchmarking machine-learning training workloads on Trn1 or Trn2 instances.
Rate: $70–$90/hour
Annualized Equivalent: $145,600–$187,200
Location: United States
Work Arrangement: Fully remote
Engagement Type: Independent contractor
Schedule: Flexible
Payment: Weekly via Stripe or Wise
Annualized compensation is based on 2,080 hours per year for comparison purposes only. Actual earnings depend on the number of hours and projects completed.
Apply your Trainium and NKI expertise to advanced AI development .
Work on technically challenging kernel evaluation and optimization problems.
Help improve AI systems' understanding of specialized accelerator architectures.
Evaluate real-world kernel migration, performance, and numerical-correctness challenges.
Contribute expertise in an increasingly important AI hardware ecosystem.
Work remotely with a flexible schedule.
Earn competitive compensation for highly specialized technical expertise.
You will be engaged as an independent contractor.
Work is fully remote and can be completed on your own schedule.
Projects may be extended, shortened, or concluded early depending on project needs and performance.
Your work will not require access to confidential or proprietary information belonging to any employer, client, or institution.
Payments are made weekly via Stripe or Wise based on services rendered.
H-1B and STEM OPT candidates cannot be supported at this time.
All qualified applicants will be considered without regard to legally protected characteristics. Reasonable accommodations are available upon request.
Similar vacancies
Recruitment Room
$145600 USD - $187200 USD / hour
Trainium (NKI) Kernel Expert Remote | Independent Contractor | United States | $145,600–$187,200 annualized ($70–$90/hour) About the Role Apply your specialized AWS Trainium and Neuron Kernel Interface (NKI) expertise to help improve the quality and reliability of advanced AI systems. As a Trainium (NKI) Kernel Expert, ...
India
$375 USD
... Linux kernel, with vast experience in performance-sensitive code and a solid understanding of the VFS, page cache & file system concepts - Familiarity with kernel development methodologies and kernel structure, as well as experience in developing kernel modules - Top-notch experience in the Linux Kernel Driver model & ...
India, Bengaluru
... Device Drivers to design, develop, integrate, and debug device drivers for embedded SoC platforms. The ideal candidate should have strong experience in Linux Kernel development, BSP development, board bring-up, hardware-software integration, and debugging embedded systems based on ARM and/or RISC-V https://jobeax.com/link/JJKfuHPEsmrFPG6l ...
India, Bengaluru
This role is for one of our clients Industry: Architecture and Planning Seniority level: Entry level Min Experience: 0+ years Location: Vadodara, Gujarat JobType: full-time
India, Vadodara
weekdayworks
This role is for one of our clients Industry: Artificial Intelligence Seniority level: Mid-Senior level Min Experience: 5+ years Location: Mumbai, Maharashtra, India JobType: full-time
India, Mumbai
HSBC Global Services Limited
... top, or simply take you in an exciting new direction, HSBC offers opportunities, support and rewards that will take you further. HSBC is one of the largest banking and financial services organisations in the world, with operations in 64 countries and territories. We aim to be where the growth is, enabling businesses to ...
India, Bengaluru
About Us: Ensono is an expert technology adviser and managed service provider. As a relentless ally, we accelerate clients’ digital transformation to achieve business outcomes that stand to last. Our dedicated team helps organizations optimize today’s systems across any hybrid environment with services such as consulting, ...
India, Pune
ZF Friedrichshafen
... doing this for more than 40 years! Today we build new drivetrain solutions to meet the requirements of the future. What you can look forward to as (Digital Expert - Full Stack Developer) : - Develop, maintain, troubleshoot, and enhance business applications and automation solutions based on business requirements, using ...
India, Coimbatore
Siemens PLM Software
377k
... software. A Linux developer with experience in embedded Linux runtime environment (Debian preferred). The ideal candidate should have recent hands-on development expertise with the major aspects of embedded Linux (embedded HW architectures, the Linux kernel, build environments, etc.). On occasion, travel may be required to meet ...
India, Bengaluru
FreshTalent
$104k USD / hour
Human Resources Expert – AI Evaluation & Subject-Matter Expert Remote | Contract | $104,000 annualized equivalent ($50/hour) About the Role Put your human resources expertise to work on projects that help improve how advanced AI systems understand and apply real-world HR principles. We’re looking for experienced Human Resources ...
India
Medcruit
$145600 USD - $239200 USD / hour
Case Manager Expert – AI Care Coordination Evaluation $145,600–$239,200/year annualized $70–$115/hour Remote | Contract Annualized equivalent based on 40 hours per week × 52 weeks. Actual earnings depend on hours worked and project availability. About the Role We’re looking for experienced case managers to apply their care ...
India
Medcruit
$135200 USD - $197600 USD / hour
... earnings depend on hours worked, project availability, and engagement needs. About the Opportunity We’re looking for experienced Medical Coders to use their expertise in clinical documentation and coding to evaluate and improve advanced AI systems. You’ll develop realistic medical coding scenarios, create expert reference ...
India
Medcruit
$83200 USD - $135200 USD / hour
... requirements. - Written assessment , if required for project placement. - Onboarding and contractor setup. - Project matching based on your EMS experience and expertise. Why Apply - Earn $40–$65/hour based on experience. - Work remotely with a flexible, asynchronous schedule. - Apply your EMS expertise to advanced AI evaluation. ...
India
genius itens
com/link/9QmyBd6n6RNe4gQ1 Mon - Fri: 9:00 AM - 09:00 PM Services Projects Apply for Digital Marketing Specialist Fields marked with * are required.
India, Surat
Medcruit
$93600 USD - $135200 USD / hour
... Complete contractor documentation and platform setup. - Project Matching — After onboarding, you may be matched with pharmacy-focused projects aligned with your expertise. Why Apply? This is an opportunity to put your pharmacy expertise to work in an emerging area of healthcare technology. You’ll contribute practical knowledge ...
India
Location: Mumbai, MH, IN Position Senior Underwriter (Expert Motor), India Corporate Title Assistant Vice President/Vice President Location Mumbai, India Reporting to Head P&C Underwriting, India About the role We have an exciting opportunity for a new role at Swiss Re, Mumbai. We seek a colleague who is passionate about ...
India, Mumbai
Recruitment Room
$104k USD - $208k USD / hour
Education & Tutoring Expert Contract · Remote · United States Annual Compensation Equivalent: $104,000–$208,000/year Hourly Rate: $50–$100/hour About the Role We are seeking experienced education and tutoring professionals to contribute to a project focused on training and improving next-generation AI systems. You will ...
India
HSBC Global Services Limited
... top, or simply take you in an exciting new direction, HSBC offers opportunities, support and rewards that will take you further. HSBC is one of the largest banking and financial services organizations in the world, with operations in 64 countries and territories. We aim to be where the growth is, enabling businesses to ...
India, Bengaluru
Medcruit
$93600 USD - $135200 USD / hour
... Complete contractor documentation and platform setup. - Project Matching — After onboarding, you may be matched with pharmacy-focused projects aligned with your expertise. Why Apply? This is an opportunity to put your pharmacy expertise to work in an emerging area of healthcare technology. You’ll contribute practical knowledge ...
India