Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference
Tasks
- Build infrastructure to analyze and onboard models
- Collaborate across compiler runtime framework and hardware teams
- Collaborate with customers to optimize ML models
- Conduct system-level performance analysis
- Deploy models using continuous deployment pipelines
- Design and develop distributed inference support
- Design and implement high performance kernels
- Implement optimization techniques such as fusion sharding tiling and scheduling
- Optimize inference latency and throughput
- Perform unit and end-to-end testing
- Profile workloads and resolve bottlenecks
- Tune machine learning models for performance
Perks/Benefits
- N/A
Skills/Tech-stack
Continuous Deployment | Deep learning | Distributed Computing | Fusion | Inference Optimization | JAX | Kernel development | Level optimization | Low-level optimization | Machine Learning | Performance Tuning | Profiling | PyTorch | Python | Scheduling | Sharding | System programming | Testing | Tiling
Education
Related jobs
-
Applied AI Engineer - AI Solutions USD 172K-300KAgentic Workflows | Airflow | Apache Spark | Chroma | CrewAIAnnual travel up to 25% | Employee stock options | Hybrid work | Professional developmentMid-level Full TimeNew York City, NY (Hybrid); Redwood … R10h ago
-
AI Solutions Engineer, Talent Acquisition USD 129K-171KAPIs | Access Control | Agentic Workflows | Audit trails | AuthenticationMid-level Full TimeSeattle, Washington, United States12h ago
-
Network Engineer, Supercomputing USD 350K-475KCUDA | Congestion Control | Container Orchestration | Debugging | Deep learningDental benefits | Health benefits | Paid parental leave | Relocation support | Unlimited PTOSenior-level Full TimeSan Francisco13h ago
-
Sr. AI Engineer USD 175K-350KAgent Orchestration | Artificial Intelligence | Backend Development | Database Design | Frontend Development401k | Dental insurance | Flexible paid time off | Home office stipend | Medical insuranceSenior-level Full TimeSanta Monica, CA14h ago
-
AI Engineer, Agentic Ad Creative (Multimodal) USD 120K-220KA/B | A/B Testing | Ad Policy Compliance | B testing | CUDA401k matching | Commuter benefits | FSA | HSA | Health, dental, and vision insuranceMid-level Full TimeMountain View, California, United States15h ago
-
Lead Principal Machine Learning Engineer USD 169K-355KAccess Control | Agent systems | AgentOps | Auditability | Autogen401k match | Life insurance | Paid Holidays | Paid parental leave | Paid sick leaveSenior-level Full TimeSan Francisco, CA, United States16h ago
-
2026 Fall Health AI Scholar, Digital Health Algorithms USD 124K-150KAWS SageMaker | Amazon Redshift | Android | Artificial Intelligence | C++Entry-level Full Time665 Clyde Avenue, Mountain View, CA, …16h ago
-
Senior-level Full TimeCosta Mesa, California, United States16h ago
-
Sr. Data Engineer (Remote) USD 163K-192KAccess Control | Amazon Web Services | Apache Iceberg | Apache Kafka | Apache Spark401k plan | Dental insurance | Disability insurance | Employee assistance program | FSA/HSASenior-level Full TimeRemote - United States R17h ago
-
Machine Learning Engineer - Simulation Framework USD 160K-234K3D Graphics | C++ | CUDA | Deep learning | Deterministic systemsEntry-level Full TimeFoster City, CA17h ago
-
Reliability Engineer, Supercomputing USD 350K-475KBMC | Container Orchestration | DCGM | Debugging | Firmware ManagementDental benefits | Health benefits | Paid parental leave | Relocation support | Unlimited PTOMid-level Full TimeSan Francisco17h ago
-
AWS | Agent Orchestration | CI/CD | Cloud platform | Databricks401k match | Counseling membership | Employer subsidized medical dental and vision | Flexible time away program | Life insuranceMid-level Full Time-REMOTE, USA- R17h ago
-
Software Engineer III - Big Data & AWS USD 175K-186KAPI Gateway | AWS Glue | AWS Lambda | AWS Step Functions | Amazon APISenior-level Full TimePlano, TX, United States17h ago
-
AI Solutions Engineer, Talent Acquisition USD 129K-171KAPI Integration | APIs | Access Control | Agentic Workflows | Audit trailsMid-level Full TimeBoston, Massachusetts, United States17h ago
-
AI Solutions Engineer, Talent Acquisition USD 129K-171KAPIs | Access Control | Agentic Workflows | Audit Logs | AuthenticationHealth insurance | Paid time off | Wellness programsMid-level Full TimeCosta Mesa, California, United States17h ago
-
Agentic Workflows | Data Pipelines | Deep learning | Deployment | Experiment designLong-term engagementMid-level Full TimeRedmond, WA17h ago
-
AI Engineer, Ecosystem USD 171K-240KAPI Integration | Access Management | Audit Logging | Authentication | AuthorizationHybrid work | Remote work up to 4 weeks per yearMid-level Full TimeSan Francisco, California, United States R17h ago
-
Forward Deployed AI Solutions Engineer USD 95K-145KAPIs | Agentic Workflows | Audit Logging | Cloud Computing | Command Line401k benefits | Commuter benefits | Employee referral program | Fertility care benefits | Free testingMid-level Full TimeUS Remote R18h ago
-
Staff Analytics Engineer USD 159K-187KClaude | Claude Code | DBT | Data Contracts | Data Modeling401k company match | Accident insurance | Company funded HSA contributions | Critical illness insurance | Health, dental, vision coverageSenior-level Full TimeRemote (United States) R18h ago
-
AI Engineer, Product USD 171K-240KA/B | A/B Testing | API Design | B testing | Data ModelingHybrid work | Remote work up to four weeks per yearMid-level Full TimeSan Francisco, California, United States R18h ago
-
AI Data Engineer USD 80K-100KAWS Bedrock | AWS CDK | Access Management | Anthropic | Authentication tokens401k | Dental insurance | Health insurance | Vision insurance | Wellness benefitsSenior-level Full TimeLos Angeles, CA, US18h ago
-
C++ | Deep learning | Distributed Training | ETL | GoSenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
C++ | ETL | Go | Hugging Face | MicroservicesFlexible work arrangements | Learning and development support | Remote work optionsSenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
Staff Machine Learning Engineer, Agentic Product USD 137K-241KAgentic AI | Generative AI | Interface Development | Language Models | Language ProcessingProfessional development support | Remote work flexibilitySenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
Abstractive summarization | Active Learning | DPO | Data Privacy | Deep learningIn-office work options | Professional development | Remote work flexibilitySenior-level Full TimeMountain View, CALIFORNIA, United States18h ago