Software Engineer II- AI/ML, AWS Neuron
Tasks
- Analyze and optimize system level performance
- Architect features for machine learning acceleration
- Collaborate with customers to enable and optimize ML models
- Deploy models using continuous deployment pipelines
- Design and implement infrastructure for model onboarding
- Develop distributed training support for PyTorch and JAX
- Implement unit and end to end model testing
- Optimize machine learning workloads across compiler runtime and hardware
- Profile systems to identify and resolve bottlenecks
- Tune machine learning models for AWS Trainium
Perks/Benefits
- N/A
Skills/Tech-stack
C++ | Continuous Deployment | Deep learning | Distributed Training | JAX | Machine Learning | Memory Optimization | Model Optimization | Performance Profiling | PyTorch | Python | System Architecture | Testing
Education
N/A
Related jobs
-
Accessibility | Artificial Intelligence | Data Analysis | Data Mining | Data ProcessingSenior-level Full TimeSunnyvale, CA, USA1h ago
-
Software Engineer III, Embedded Systems Firmware, Platforms Infrastructure Engineering USD 147K-211KC# | C++ | Data Structures | Data Structures and Algorithms | Device DriversSenior-level Full TimeSunnyvale, CA, USA1h ago
-
Staff Software Engineer, AI/ML Performance USD 207K-300KAlgorithm Design | Auto sharding | C++ | Code generation | Compiler optimizationSenior-level Full TimeSunnyvale, CA, USA1h ago
-
Technical Lead Manager, ML and GenAI, Ads Safety USD 207K-300KAgentic AI | Artificial Intelligence | Context learning | Distributed Computing | EmbeddingsSenior-level Full TimeMountain View, CA, USA; New York, …1h ago
-
AI Middleware Engineer USD 99K-195KAI Agent | AI Agent Frameworks | API Design | API Security | Agent FrameworksClient-facing work | Production impactMid-level Full TimeSan Francisco8h ago
-
Software Engineer, Applied AI USD 130K-500KData Pipelines | Data Quality | Evaluation | Experimental Design | GoEquity 4 year vest | Free gym membership | Health insurance | Housing bonus | Meal stipendMid-level Full TimeSan Francisco9h ago
-
Software Engineer, Storage USD 157K-185KAsynchronous processing | CDC | Change Data Capture | Data Capture | Data IngestionEmployee travel credits | Oncall rotation | Remote eligibleSenior-level Full TimeUnited States10h ago
-
Senior Data Engineer USD 136K-160KApache Spark | Azure Data | Azure Data Factory | Azure Event | Azure Event Hubs401k match | Employee assistance program | Free parking | Healthcare | Hybrid workplace flexibilitySenior-level Full TimeCharlotte, North Carolina, United States; Virtual R10h ago
-
Optimization Software Engineer USD 132K-198KAPIs | AWS | Algorithms | Amazon CodeWhisperer) | Azure401k matching | Commuter benefits | Healthcare benefits | Life and disability insurance | Mental health resourcesMid-level Full TimeWashington, District of Columbia, United States10h ago
-
Quantum Software Engineer USD 132K-198KC# | C++ | CI/CD | CUDACUDAQ | CUDALabCaregiver leave | Commuter benefits | Disability insurance | Family planning support | Life insuranceMid-level Full TimeWashington, District of Columbia, United States10h ago
-
Mid-level Full TimeDes Moines, Iowa12h ago
-
Senior-level Full TimeNew York, NY, United States12h ago
-
Robotic Software Engineer - Air Autonomy USD 135K-175KAgile | ArduPilot | Bash | C++ | CI/CDIn-office work flexibility | Occasional remote work | Paid time off | Travel opportunitiesMid-level Full TimeWashington, District of Columbia, United States13h ago
-
Robotic Software Engineer - Aerial Autonomy USD 125K-165KAgile | ArduPilot | Bash | C++ | CI/CDTravel for site integration and demonstrationsMid-level Full TimeFort Walton Beach, Florida, United States13h ago
-
Robot Learning Engineering Intern USD 80K-116KBehavior Cloning | Computer Vision | Force control | Imitation Learning | Learning from DemonstrationCatered lunches | Employee Assistance Program (EAP) | Flexible PTO | Healthy snacks | Paid sick leaveEntry-level InternshipOnsite- Pittsburgh, PA14h ago
-
Product Enablement Data Engineer USD 112K-182KAWS | Authentication | Azure | Curl | DNS401k | Dental insurance | Disability benefits | Flexible time off | Health insuranceSenior-level Full TimeUSA14h ago
-
AI Data Engineer USD 133K-256KAWS | Agile | Apache Kafka | Apache Spark | AzureHybrid work | Work from home up to three days per weekEntry-level Full TimeMaryland, United States14h ago
-
Sr Machine Learning Engineer I USD 150K-241KAutomated retraining | Bayesian Filter | CI/CD | Cloud Computing | Data AnalysisDiscretionary paid time off | Emotional and mental wellness support | Employee resource groups | Fitness programs | Learning and development programsSenior-level Full TimeSeattle, Washington, United States14h ago
-
AWS Lambda | Amazon Bedrock | Amazon ECR | Amazon S3 | Amazon SageMaker401k match | Dental insurance | Health insurance | Long-term engagement | Remote-firstSenior-level Full TimeUSA15h ago
-
BACnet | BMS | Control Systems | Control loops | Cooling systemsFlexible working hours | Generous vacation | Optional equity donation matching | Parental leave | Visa sponsorship assistanceMid-level Full TimeRemote-Friendly, United States R15h ago
-
AI/ML Research Engineer USD 120K-220KAWS | Cloud Computing | Computational Biology | Data Processing | Data pipelineMid-level Full TimeBoston15h ago
-
IT AI&S Senior Data Engineer USD 132K-202KAWS | Apache Spark | Azure | Data Governance | Data LineageSenior-level Full TimeBoston, MA, United States15h ago
-
Senior/Lead Bioinformatics Engineer USD 133K-166KAWS | CI/CD | Containerization | Distributed Computing | Docker401k benefits | Commuter benefits | Comprehensive medical/dental/vision coverage | Employee referral program | Fertility care benefitsSenior-level Full TimeUS Remote R15h ago
-
Senior Software Engineer, AI Solutions USD 120K-142KAI Agents | API Integration | AWS Bedrock | AWS SageMaker | Airflow403b retirement plan | Counseling support | Dental insurance | Dependent Care Account | Disability LeaveSenior-level Full TimeNew York, NY, United States15h ago
-
AWS | Agile | Azure | Blob Storage | C#Benefits package | Equity compensation | Onsite work scheduleSenior-level Full TimeSan Mateo, CA, United States16h ago