Software Engineer I - AI/ML, AWS Neuron Distributed Training
Tasks
- Develop distributed training solutions
- Extend distributed training frameworks
- Manage gradients for stability
- Optimize low precision training
- Optimize mixed precision training
- Profile end to end training performance
- Support deployment of training workloads
- Tune training pipelines
Perks/Benefits
- N/A
Skills/Tech-stack
AWS Trainium | BF16 | Deep learning | Distributed Training | FP8 | FSDP | Gradient Management | Hugging Face | Language Models | Large Language Models | Loss Scaling | Machine Learning | Mixture of Experts | Performance Profiling | PyTorch | Python | Reinforcement Learning | Reinforcement Learning Workloads | TorchTitan | Training Optimization | Transformers
Education
Regions
Countries
States
Cities
Related jobs
-
Software Engineer, ML Data Infrastructure USD 160K-220KAlgorithms | Cloud platform | Data Pipelines | Data Processing | Data StorageAccess to compute resources | Dental insurance | Health insurance | In-office meals | Learning and mentorshipEntry-level Full TimeToronto4h ago
-
Computer Vision AI & ML Engineer USD 100K-300K3D Geometry | 3D Scene | 3D Scene Understanding | Computer Vision | Data AugmentationEntry-level Full TimeSan Mateo, CA8h ago
-
Autonomy Algorithm Engineer, Planning & Prediction USD 140K-225KAgent modeling | C++ | GPU Inference | HD Mapping | Imitation LearningMid-level Full TimeHouston, TX or San Francisco Bay …9h ago
-
Algorithm Engineer, Autonomy Planning & Prediction USD 150K-200KC++ | CI/CD | Code review | Data Engineering | DebuggingMid-level Full TimeHouston, TX or San Francisco Bay …9h ago
-
Amazon Redshift | Data Modeling | Data Pipelines | ETL | ExperimentationMid-level Full TimeBellevue, Washington, USA10h ago
-
Senior AI Engineer USD 216K-277KAWS | Agent systems | Autonomous Agents | Azure | By DesignCareer development | Home workspace stipend | MacBook provided | Medical, dental, and vision insurance | Mental health benefitsSenior-level Full TimeAustin, TX R10h ago
-
AI Engineer at Lumion USD 148K-200KAPI Design | Data Modeling | Debugging | Guardrails | Language Models401k | Health dental vision reimbursement | MacBook Pro | Relocation support | Tech stipendMid-level Full TimeSouth Jordan, UT, US, CA, US11h ago
-
Enterprise AI Platform Engineer USD 130K-150KAPI Gateway | AWS | AWS IAM | Audit Logging | Azure401k | Dental insurance | Flexible PTO | Growth opportunities | Medical insuranceSenior-level Full TimeAustin, TX11h ago
-
Software Engineer, Storage USD 140K-265KAlerting | Autoscaling | C++ | Cloud Native | Databases401k contribution | Dental insurance | Education stipend | Healthy lunches | Home office improvement stipendMid-level Full TimeMountain View, CA11h ago
-
Senior Data Engineer USD 135K-165KAmazon Kinesis | Amazon Redshift | Apache Airflow | Apache Iceberg | Apache Kafka401k | Healthcare benefits | Paid time offSenior-level Full TimeVirtual, United States R12h ago
-
Software Engineer II, Machine Learning USD 145K-175KAWS | Airflow | CI/CD | Data Preparation | Experimentation401k employer match | Commuter subsidy | Concierge medical membership | Dental insurance | Fitness membership subsidyMid-level Full TimePalo Alto, California12h ago
-
AI Architect - Business Applications USD 138K-225KA/B | A/B Testing | API | B testing | ExperimentationHybrid work optionSenior-level Full TimeBellevue, WA, United States12h ago
-
Staff Data Engineer, Finance USD 170K-230KAWS | Asset-based lending | DBT | Data Ingestion | Data Modeling401k matching | Dental insurance | Employee assistance program | Employee recognition | Employee referral programSenior-level Full TimeSeattle, Washington, United States13h ago
-
Junior Data Engineer USD 64K-90KDatabricks | ELT | ETL | Git | JiraActive Secret security clearanceEntry-level Full TimeArlington, VA13h ago
-
Staff Data Engineer USD 180K-200KAWS | Airbyte | Airflow | Apache Iceberg | Automated testing401k match | Dental insurance | Disability insurance | Employee assistance program | Life insuranceSenior-level Full TimeWaltham, MA13h ago
-
Staff Data Engineer USD 180K-200KAPI Development | AWS | Airbyte | Airflow | Amazon Kinesis401k match | Dental insurance | Disability insurance | Employee assistance program | Health and wellness resourcesSenior-level Full TimeNorth Bethesda, MD13h ago
-
AWS | B-Rep | C++ | CAD interoperability | CMakeConference participation | Design review leadership | Mentorship | Technical leadership opportunitiesSenior-level Full TimeWaltham, MA13h ago
-
ACIS | AWS | B-Rep | C++ | CAESenior-level Full TimeNorth Bethesda, MD13h ago
-
Senior Embedded Firmware Engineer USD 120K-153KADC | Bare Metal | C plus plus | C# | Control AlgorithmsOwnership equity | Work-life balanceSenior-level Full TimeWilmington, MA14h ago
-
Data Engineer USD 170K-220KApache Airflow | Apache Beam | Apache Flink | Apache Spark | BigQueryComprehensive benefitsSenior-level Full TimeSan Francisco or Denver14h ago
-
Principal AI Engineer, AI Solutions USD 172K-216KAPI Design | AWS | Agent Frameworks | Automation | Data PipelinesCareer Development Programs | Commuting cost coverage | Daily free lunch | Employee resource groups | Equity grantsSenior-level Full TimeBoston, Massachusetts, United States R14h ago
-
Senior Data Engineer USD 150K-228KApache Flink | Apache Spark | BigQuery | Data Architecture | Data QualityOn-site work flexibilitySenior-level Full TimeChicago, Illinois14h ago
-
Senior Data Engineer USD 165K-216KAWS | Apache Airflow | BI Integration | CI/CD | Cloud Computing401k match | Company events | Dental insurance | Employee discount | Free swagSenior-level Full TimeBeverly Hills14h ago
-
Senior Machine Learning Engineer USD 129K-200KC++ | CUDA | Camera Pose Estimation | Computer Vision | GPU ProgrammingSenior-level Full TimeSunnyvale, CA, United States14h ago
-
Staff Machine Learning Engineer USD 200K-220KAWS | Alerting | Auto Scaling | Automated retraining | CI/CD401k match | Dental insurance | Disability insurance | EAP support | Life insuranceSenior-level Full TimeWaltham, MA15h ago