AWS Trainium & Neuron Kernel Interface (NKI) Technical Specialist (Freelance Contract)

Project Overview: We are sourcing experienced technical specialists in AWS Trainium and Neuron Kernel Interface (NKI) to audit tasks used to train and evaluate AI systems. The objective of this project is to ensure that AI training workflows and tasks are technically rigorous, practical, and highly accurate within your specific area of expertise.

Project Deliverables & Scope

• Task Evaluation: Assess whether AWS Trainium/NKI tasks are technically accurate, realistic, solvable, reproducible, and supported by reliable tests and evaluation criteria.
• Technical Auditing & Feedback: Provide clear, actionable feedback on any identified hardware acceleration inefficiencies, kernel execution bugs, or logic errors within the tasks.

Required Expertise

• Demonstrable professional experience and deep technical knowledge in AWS Trainium architecture (including model compilation, accelerator optimization, and NKI frameworks). (Note: Candidates will be considered for one specialty based on their experience; expertise across other domains is not required).
• Strong analytical and problem-solving skills to rigorously test, troubleshoot, and evaluate complex technical scenarios.