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
-
Data Engineer USD 192K-196KBig Data | Data Architecture | Data Visualization | Data Warehouse | Database systemsEntry-level Full TimeMenlo Park, CA3h ago
-
Data Engineer, Analytics USD 209K-235KBig Data | Data Governance | Data Modeling | Data Quality | Data VisualizationSenior-level Full TimeMenlo Park, CA3h ago
-
Data Engineer, Analytics USD 185K-196KBig Data | Data Governance | Data Quality | Data Security | Data WarehousingMid-level Full TimeMenlo Park, CA3h ago
-
Senior-level Full TimeNew York, NY3h ago
-
Accessibility | Artificial Intelligence | Data Analysis | Data Mining | Data ProcessingSenior-level Full TimeSunnyvale, CA, USA3h 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, USA3h ago
-
Staff Software Engineer, AI/ML Performance USD 207K-300KAlgorithm Design | Auto sharding | C++ | Code generation | Compiler optimizationSenior-level Full TimeSunnyvale, CA, USA3h 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, …3h ago
-
API Integration | Cloud Architecture | Data Processing | Deep learning | GPUTravel up to 20 percent timeSenior-level Full TimeReston, VA, USA; Washington D.C., DC, …3h ago
-
AI Middleware Engineer USD 99K-195KAI Agent | AI Agent Frameworks | API Design | API Security | Agent FrameworksClient-facing work | Production impactMid-level Full TimeSan Francisco9h 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 Francisco11h 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 States11h 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 R12h 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 States12h 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 States12h ago
-
Mid-level Full TimeDes Moines, Iowa14h ago
-
Senior-level Full TimeNew York, NY, United States14h 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 States15h 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 States15h 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, PA16h ago
-
Product Enablement Data Engineer USD 112K-182KAWS | Authentication | Azure | Curl | DNS401k | Dental insurance | Disability benefits | Flexible time off | Health insuranceSenior-level Full TimeUSA16h 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 States16h 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 States16h ago
-
AWS Lambda | Amazon Bedrock | Amazon ECR | Amazon S3 | Amazon SageMaker401k match | Dental insurance | Health insurance | Long-term engagement | Remote-firstSenior-level Full TimeUSA16h 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 R17h ago