Applying for AWS Trainium / NKI Kernel Expert at Anyone AI? Compare this live job with your resume first.

🎯 Tailor my resume
←Jobiba.Me
AA
Active1mo ago

AWS Trainium / NKI Kernel Expert

Anyone AI·📍 Buenos Aires, Distrito Federal, Argentina

Employment

CONTRACT

Work mode

Remote

Experience

2+ years

Salary

$65/hr

Role overview

Job description

Anyone AI is recruiting experienced AWS Trainium / Neuron Kernel Interface (NKI) engineers for a specialized project focused on evaluating and improving kernel development tasks for AI workloads. We’re looking for engineers with hands-on experience building or optimizing NKI kernels on AWS Trainium or Inferentia2 hardware who understand how Trainium’s architecture differs from traditional GPU programming. What You’ll Work On You’ll review and evaluate technical tasks involving: NKI kernel correctness and Trainium-specific development patterns CUDA → NKI kernel migrations Trainium performance optimization and benchmarking Memory management across SBUF, PSUM, and HBM Tile-based computation and DMA scheduling Cross-platform numerical correctness between CUDA/Triton and NKI Trainium-specific performance bottlenecks and optimization opportunities Technical feedback and quality assessment of k

…

About this listing

  • Source: Anyone AI's official careers site (via Ashby) · last confirmed on the source 29 Oct 2026
  • Posted 15 Sept 2026

Jobiba does not charge job seekers. Never pay to apply or share OTPs, card or bank details.

What they are looking for

Skills & requirements

awsai

Department · Software Engineering

Hiring company

Anyone AI

Finance & Banking

El AI Fellowship de Anyone AI es un programa de 4 meses, 100% remoto y en vivo, para pasar de Software Developer a AI Engineer con clases en vivo.

Related opportunities

Similar active roles

Relevant, currently active roles for this market.

All jobs →

🔔 Get job alerts

New AWS Trainium jobs in Buenos Aires, Distrito Federal, Argentina, straight to your inbox.

No spam · Unsubscribe anytime