Software Engineer

15 hours ago


Cupertino CA United States Amazon Full time

Job ID: 2718972 | Amazon Web Services, Inc. - A97

AWS Neuron is the complete software stack for the AWS Inferentia (Inf1/Inf2) and Trainium (Trn1), our cloud-scale Machine Learning accelerators. This role is for a machine learning engineer in the Distribute Training team for AWS Neuron, responsible for development, enablement and performance tuning of a wide variety of ML model families, including massive-scale Large Language Models (LLM) such as GPT and Llama, as well as Stable Diffusion, Vision Transformers (ViT) and many more.

The ML Distributed Training team works side by side with chip architects, compiler engineers and runtime engineers to create, build and tune distributed training solutions with Trainium instances. Experience with training these large models using Python is a must. FSDP (Fully-Sharded Data Parallel), Deepspeed and other distributed training libraries are central to this and extending all of this for the Neuron based system is key.

Key job responsibilities
You will help lead the efforts building distributed training support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. You will help tune these models to ensure highest performance and maximize the efficiency of them running on the custom AWS Trainium and Inferentia silicon and the Trn1, Inf1/2 servers. Strong software development and Machine Learning knowledge are both critical to this role.

About the team

Annapurna Labs was a startup company acquired by AWS in 2015, and is now fully integrated. If AWS is an infrastructure company, then think Annapurna Labs as the infrastructure provider of AWS. Our org covers multiple disciplines including silicon engineering, hardware design and verification, software, and operations. AWS Nitro, ENA, EFA, Graviton and F1 EC2 Instances, AWS Neuron, Inferentia and Trainium ML Accelerators, and in storage with scalable NVMe, are some of the products we have delivered, over the last few years.

Inclusive Team Culture

Here at AWS, we embrace our differences. We are committed to furthering our culture of inclusion. We have ten employee-led affinity groups, reaching 40,000 employees in over 190 chapters globally. We have innovative benefit offerings, and host annual and ongoing learning experiences, including our Conversations on Race and Ethnicity (CORE) and AmazeCon (gender diversity) conferences. Amazon’s culture of inclusion is reinforced within our 16 Leadership Principles, which remind team members to seek diverse perspectives, learn and be curious, and earn trust.

Work/Life Balance

Our team puts a high value on work-life balance. It isn’t about how many hours you spend at home or at work; it’s about the flow you establish that brings energy to both parts of your life. We believe striking the right balance between your personal and professional life is critical to life-long happiness and fulfillment. We offer flexibility in working hours and encourage you to find your own balance between your work and personal lives.

Mentorship & Career Growth

Our team is dedicated to supporting new members. We have a broad mix of experience levels and tenures, and we’re building an environment that celebrates knowledge sharing and mentorship. We care about your career growth and strive to assign projects based on what will help each team member develop into a better-rounded professional and enable them to take on more complex tasks in the future.

BASIC QUALIFICATIONS

- Bachelor's degree in computer science or equivalent
- 3+ years of non-internship professional software development experience
- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience programming with at least one software programming language
- Experience in machine learning, data mining, information retrieval, statistics or natural language processing

PREFERRED QUALIFICATIONS

- Master's degree in computer science or equivalent
- 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Experience in computer architecture
- Previous software engineering expertise with Pytorch/Jax/Tensorflow, Distributed libraries and Frameworks, End-to-end Model Training.
- Previous experience with training multi-modal models for understanding and generating images/videos/audios

Posted: July 3, 2024 (Updated about 14 hours ago)

#J-18808-Ljbffr

  • Cupertino, United States Lynx Software Technologies Full time

    Thompson Software Solutions is seeking a mid to senior level Software Engineer who is ready to work with a talented team to provide innovative solutions for tomorrows problems. This position requires a software engineer to use a wide application of technical principles, theories, and concepts in the software field to develop, integrate, and test software...


  • Cupertino, United States Lynx Software Technologies Full time

    Thompson Software Solutions is seeking a mid to senior level Software Engineer who is ready to work with a talented team to provide innovative solutions for tomorrows problems. This position requires a software engineer to use a wide application of technical principles, theories, and concepts in the software field to develop, integrate, and test software...

  • Software Engineer

    2 days ago


    Los Angeles, CA, United States Software Resources Full time

    We are looking for Software Engineer for our client located in West Los Angeles/Century City area.This position is mostly REMOTE but you will be required to come into the West Los Angeles office 1 to 4 times a month. We could consider a completely remote candidate in another state if they meet our requirements. Duration : 6 to 12 months with the potential to...

  • Software Engineer

    16 hours ago


    Cupertino, CA, United States Apple Full time

    Software Engineer Cupertino, California, United States Hardware The Satellite Connectivity Group is seeking a senior engineer experienced in software development for a distributed production system, as well as for data transformation, visualization and tooling. You will collaborate across our group to implement new services and functionality for our...

  • Software Engineer

    16 hours ago


    Cupertino, CA, United States Apple Inc. Full time

    The Satellite Connectivity Group is seeking a senior engineer experienced in software development for a distributed production system, as well as for data transformation, visualization, and tooling. You will collaborate across our group to implement new services and functionality for our network. You should have strong technical skills, be able to think...


  • San Jose, CA, United States Innominds Software Pvt. Ltd. Full time

    Engage in requirements gathering and design discussions for product development. Collaborate with stakeholders to define test objectives, assess testing scope, and formulate functional and non-functional test strategies, plans, and artifacts. Develop comprehensive test cases for new product requirements. Create and manage test epics, user stories, tickets,...


  • Sunnyvale, CA, United States Software Aspekte Full time

    Fortinet is looking for a Senior Software Dev QA Engineer to assist CloudServices Team to Maintain and develop automated test scripts to validate all aspect of products based on variety of test cases. Provide end-to-end quality assurance solutions for cloud security products to the development, marketing, sales team etc. Bring cutting-edge cloud security...


  • Omaha, NE, United States Election Systems & Software Full time

    We support democracy.Our vision at Election Systems & Software (ES&S) is simple and unwavering — we provide products and services of exceptional quality and value to maintain voter confidence and enhance the voting experience. We accomplish this through continuous investments in research and development, resulting in regular product updates and security...


  • Omaha, NE, United States Election Systems & Software Full time

    We support democracy.The information below covers the role requirements, expected candidate experience, and accompanying qualifications.Our vision at Election Systems & Software (ES&S) is simple and unwavering — we provide products and services of exceptional quality and value to maintain voter confidence and enhance the voting experience. We accomplish...


  • Omaha, NE, United States Election Systems & Software Full time

    Job DescriptionJob DescriptionDescription:We support democracy.Any additional information you require for this job can be found in the below text Make sure to read thoroughly, then apply.Our vision at Election Systems & Software (ES&S) is simple and unwavering — we provide products and services of exceptional quality and value to maintain voter confidence...


  • Huntsville, AL, United States Thompson Software Solutions Full time

    Thompson Software Solutions, a defense-focused software company of proven and dedicated employees is seeking a senior-level Embedded Software Engineer who is ready to work with a talented team to provide innovative solutions for tomorrow’s problems.This position requires a software engineer to use a wide application of technical principles, theories, and...


  • Cupertino, CA, United States Apple Inc. Full time

    Apple Silicon Engineering is seeking Hardware-minded Software engineers to build innovative tools to design the next generation of Apple’s world-leading systems-on-chip (SOCs). Apple’s multi-billion-transistor Ax SOCs are the brain of every iPhone and iPad. The Analog Mixed-Signal (AMS) circuits connect them to the physical world via complex IPs such as...


  • Redmond, WA, United States Intelliswift Software Full time

    Job Title: Robotics Software Engineer VLocations: 100% Onsite at Redmond, WA 98052Duration: 12 Months Contract on W2 (Possible Extension)Summary:We are seeking a highly skilled software engineer to work on hardware integration and automation projects. The ideal candidate will have experience working with consumer electronics, firmware, and lab test equipment...


  • Cupertino, CA, United States Apple Inc. Full time

    Embedded Software Engineer - Biophotonics Imagine what you could do here! At Apple, new ideas have a way of becoming extraordinary products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no telling what you could accomplish. Dynamic, smart people and inspiring, innovative technologies are the norm here....


  • Atlanta, GA, United States Ultimate Software Full time

    Here at UKG, our purpose is people. Our HR, payroll, and workforce management solutions help organizations unlock happier outcomes for all. Our U Krewers, who build those solutions and support our business, are talented, collaborative, and innovative problem-solvers. We strive to create a culture of belonging and an employee experience that empowers our...


  • Cupertino, CA, United States Apple Inc. Full time

    The SPEAR team in Apple’s Security Engineering & Architecture organization is hiring a collaborative software engineer with a drive to deploy Swift in operating system software to improve the security of end user devices. SPEAR is a dedicated team of software engineers with a passion for user security. We work hands-on with teams across Apple to improve...


  • Cupertino, CA, United States Apple Inc. Full time

    Apple's camera algorithm team is looking for extraordinary software developers to work on an image capture and processing/rendering prototyping platform that serves camera algorithm development on all Apple product cameras. As part of the team, you will work on camera, ISP, Machine Learning technologies based on Apple-crafted image signal processing pipeline...

  • Software Engineer

    2 days ago


    Cupertino, CA, United States Apple Inc. Full time

    Join the team responsible for Apple’s Beta Software Program. Our app, Feedback Assistant, collects user feedback on beta software releases, while our internal tools analyze, group, and send meaningful feedback to Apple's engineering teams. We are looking for engineers to drive the modernization of our backend infrastructure, develop tools and automation...


  • Cupertino, CA, United States Apple Full time

    Power Management Software Engineer Cupertino, California, United States Software and Services Are you a big-picture thinker who loves setting ambitious goals? Do you have a passion for understanding how each line of code affects all the others? In the Core Operating Systems group ensuring the OS is inseparable from each device’s identity as a whole....


  • Redmond, WA, United States Talent Software Services Full time

    Software Design Engineer 2 Make sure to apply quickly in order to maximise your chances of being considered for an interview Read the complete job description below. Job Summary: Talent Software Services is in search of a Software Design Engineer for a contract position in Redmond, WA. The opportunity will be seven months with a strong chance for a...