Senior Python Developer - Artificial Intelligence & Machine Learning
We're seeking a highly skilled Senior Python Developer to join a major Federal Government research and innovation program delivering advanced AI and machine learning applications.
You'll lead the design, development, and optimisation of machine learning models - transforming complex research into production-ready systems that deliver meaningful real-world impact. Working closely with data scientists, engineers, and researchers, you'll help shape the future of AI-enabled solutions in Australia.
About the Opportunity
Join a forward-thinking Federal Government initiative focused on applied AI, machine learning, and data science. You'll be part of a collaborative Agile team developing high-performance models and applications using cutting-edge tools like PyTorch, TensorFlow, Hugging Face, and Ultralytics.
This role is ideal for someone who thrives in a technical leadership capacity - balancing innovation, delivery, and mentorship while ensuring quality and scalability in every solution.
Key Responsibilities
Design, develop, and optimise machine learning models and data-driven applications.
Translate research into production by implementing algorithms from academic papers.
Build and fine-tune AI models using PyTorch, TensorFlow, Hugging Face, or Ultralytics.
Conduct data preprocessing, annotation, and augmentation for model development.
Integrate AI solutions into existing enterprise workflows via REST APIs and CI/CD pipelines.
Manage Agile project tasks using Jira, GitLab, or GitHub Projects.
Evaluate and benchmark model performance using metrics and industry tools.
Collaborate with research and engineering teams to deliver scalable AI systems.
Essential Skills & Experience
5+ years' experience in software development with Python.
Strong background in AI/ML model development and deployment.
Proven experience with MLOps, version control, and CI/CD frameworks.
Understanding of AI model architectures such as Transformers, CNNs, and LLMs.
Strong communication skills and ability to translate technical outcomes to stakeholders.
Desirable Experience
Deep Learning: PyTorch, TensorFlow
ML Libraries: NumPy, Hugging Face, Ultralytics (YOLO/DETR)
MLOps Tools: Weights & Biases (W&B), MLflow
API Development: FastAPI, Flask, Django
Containerisation: Docker, Kubernetes
Workflow Orchestration: Airflow, Prefect, NiFi
Agile/Scrum delivery experience
The Essentials
Eligibility: Australian citizenship but Negative Vetting Level 1 is required
Location: Melbourne location - onsite
Engagement: 12 month contract starting in the new year
Remuneration: Competitive hourly rate based on experience
Why Join?
Work on nationally significant AI and ML initiatives
Collaborate with top-tier engineers, researchers, and data scientists
Be part of an organisation shaping the future of intelligent technology
