Amazon32 followers154 jobs

Machine Learning Engineer, Generative AI Innovation Center

Mid levelFull-timeHybridMelbourne VIC, AustraliaDataInformation technologySoftware engineeringPosted 1 month ago
View your fit0 of 9 criteria met

About the job

As a Software Development Engineer (SDE) at the Generative AI Innovation Center at AWS, you will embark on an exciting journey to develop custom Large Language Models (LLMs) that empower businesses with transformative AI solutions. Your work will have a significant impact on optimizing AI models for high-performance deployment, enabling customers to harness cutting-edge technology. Collaborating with a diverse team of experts, you will thrive in an inclusive culture that values innovation and continuous learning.

You'll be responsible for

🔧

Designing and implementing training pipelines

Design and implement distributed training pipelines for Large Language Models (LLMs) using tools such as Fully Sharded Data Parallel (FSDP) and DeepSpeed, ensuring scalability and efficiency.
🎨

Customizing and fine-tuning LLMs

Adapt LLMs for new languages, domains, and vision applications through continued pre-training, fine-tuning, and Reinforcement Learning with Human Feedback (RLHF).
🚀

Optimizing models for AWS

Optimize AI models for deployment on AWS's custom AI accelerators, enhancing performance and efficiency.

Skills you'll need

🤖

Deep learning and machine learning methods

Hands-on experience with deep learning and/or machine learning methods for training, fine-tuning, and inference.
🧠

Generative AI technology

Hands-on experience with generative AI technology, including developing, deploying, or optimizing machine learning models.
💻

Software development life cycle

Experience with the full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations.
View more

Meet the team

About the company

View your fit

0 of 5 criteria met
3+ years of software development experience
Proven experience in professional software development is essential.

View

Hands-on experience with deep learning
Experience with training, fine-tuning, and inference methods is required.

View

Experience with generative AI technology
Demonstrated knowledge in generative AI applications is crucial.

View

1+ years of machine learning model optimization
Experience with recognized ML libraries for model deployment is needed.

View

2+ years in system design or architecture
Experience in designing reliable and scalable systems is important.

View

A meaningful career starts with a match

  • Explore
    • Jobs
    • Companies
    • People
    • Hatch Hotlist 2025
  • Hiring
    • Permanent hires
    • Hatch Placements
  • Resources
    • Blog
    • Community stories
    • Career advice
    • Customer stories
    • Help centre
  • Hatch
    • About
    • Careers
    • Contact
    • Hatch updates
    • Media enquiries
  • © 2025 Hatch
  • Privacy
  • Terms

Similar jobs

View all
REA Group
REA Group
Machine Learning EngineerRichmond, VIC · Full-time
This job is about making a significant impact in a collaborative and supportive team focused on AI. Working with a leading property brand, you will have the opportunity to influence millions of users. The team thrives on innovation and high performance, creating a vibrant culture that values care and collaboration.
TikTok
TikTok
Machine Learning Engineer - Trust and Safety, NLP (Australia)Sydney, NSW · Full-time
This job is about building a safer online community at TikTok through innovative Trust & Safety initiatives. By designing and deploying advanced machine learning models, you will play a crucial role in ensuring that users can discover, create, and connect safely. The team thrives on collaboration, creativity, and a shared passion for tackling challenging problems together.
Amazon
Amazon
Machine Learning Scientist, International Machine LearningMelbourne, VIC · Full-time
This job offers an exciting opportunity to join Amazon's International Machine Learning team in Melbourne, where talented Machine Learning Scientists collaborate to innovate and enhance customer experiences globally. By working on high-impact projects, you will push the boundaries of applied science, directly benefiting millions of customers. The team thrives on creativity and collaboration, valuing contributions to both practical applications and the research community.