Software Development Engineer, Neuron Explorer
Core
Design, build, and operate the full stack (backend, web UI, CLI) for Neuron Explorer, a tool for profiling, debugging, and optimizing machine learning workloads on AWS Trainium and Inferentia hardware.
Role type
Senior Software Development Engineer (Full Stack)
Builds
Neuron Explorer tooling including performance data pipelines, interactive visualizations, and CLI for ML workload optimization
Domain
Cloud-scale machine learning accelerators / AI infrastructure
Deliverable
production ML models | product features
Required skills
Full software development lifecycle, system design and architecture, performance data analysis, backend development, web UI development, CLI development, generative AI integration
Preferred skills
PyTorch or JAX, Machine Learning and LLM fundamentals, deploying LLMs on AI acceleration hardware
Technologies
AWS Trainium, AWS Inferentia, Neuron, generative AI
Responsibilities
Design and build performance data collection pipelines and analysis engines; Develop interactive visualizations and developer workflows; Own features end to end across the full software development lifecycle; Collaborate with compiler, runtime, and chip architecture teams; Evaluate and integrate generative AI capabilities into the recommendation engine
Seniority
Senior, hands-on IC