Graduate Machine Learning Engineer, Remote

Who we are

We are a small, self-funded team building a booking app for independent salons and tattoo studios. We started in 2022 and have grown to about 48 people. Most engineers work remotely and come into the office one week a quarter for planning.

About the role

Western Technologies has an opening for a Graduate Machine Learning Engineer, Remote. The operations team trains and serves models that run on every customer request. We need someone who cares about latency as much as accuracy. This hire will help build out our ML platform: feature store, training pipelines, model registry and deployment. Our team is spread across 9 countries. Headquarters is in Melbourne. We engage team members in India as full-time contractors. Most of our contractors have been with us for more than two years. Core hours overlap with Sydney business hours until mid-afternoon. Day-to-day communication is in Slack and Google Meet.

What you'll do

• Productionize data science prototypes without rewriting everything.
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow.
• Track experiments, artifacts and versions in MLflow or Weights & Biases.
• Build training pipelines in PyTorch or TensorFlow and run them on GPUs in AWS, GCP or Azure.
• Handle on-call for the ML services the team owns.
• Fine-tune open-weight models with LoRA when it beats prompting.
• Develop RAG pipelines with embeddings, vector stores (pgvector, Pinecone, Weaviate) and rerankers.

What you bring

• 1+ years of software engineering experience, with at least two spent on ML systems in production.
• BS or MS in computer science, engineering or a related field.
• Strong Python and solid engineering habits (testing, packaging, code review)
• Understanding of core ML concepts: overfitting, evaluation metrics, bias in training data.

Pluses

• Recommendation or ranking systems at scale.
• Distributed training with DeepSpeed or FSDP.
• Published papers or open source ML contributions.

What we offer

Salary range: $1,850 - $2,150/month

• Monthly payments via wire transfer or Wise.
• Paid time off.
• Flexible hours.

Duration: 6 months. Start: Immediately.

Demo job created on 11jobs for testing and demonstrations. It is not an actual opening at Western Technologies Pty Ltd.