Machine Learning Engineer

Since 2013, the company has expanded from listings into payments, shipping and seller financing. More than 780 employees work on the product, from search ranking to fraud detection. Core hours are 10 to 3, and the rest of the day is flexible.

About the role

Juneberry Commerce is hiring for the Machine Learning Engineer role to take models from notebook to production and keep them working once they're there. We're looking for an engineer to build retrieval-augmented generation features on hosted and open-weight language models, with proper evaluation. This role is fully remote within Australia. We keep meetings to a minimum and work mostly async. You get a company laptop and a headset before you start.

Day to day

• Develop RAG pipelines with embeddings, vector stores (pgvector, Pinecone, Weaviate) and rerankers
• Write clean, tested Python and review teammates' code
• Build training pipelines in PyTorch or TensorFlow and run them on GPUs in AWS, GCP or Azure
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow
• Fine-tune open-weight models with LoRA when it beats prompting
• Deploy models behind low-latency APIs with FastAPI, Triton, TorchServe or vLLM

Required qualifications

• 3-5 years of software engineering experience, with at least two spent on ML systems in production
• Knowledge of LLM APIs, prompt design and evaluation methods
• Experience deploying models as services and monitoring them
• Comfortable with SQL and large-scale data processing in Spark or Ray
• Familiarity with Docker, Kubernetes and a cloud ML platform (SageMaker, Vertex AI, Azure ML)
• Hands-on experience with PyTorch, TensorFlow or JAX

Nice to have

• Published papers or open source ML contributions
• CUDA or Triton kernel experience
• Distributed training with DeepSpeed or FSDP

$183,000 - $215,000 per annum

What we offer

• Superannuation on top of salary
• 17.5% leave loading
• Annual bonus

Aboriginal and Torres Strait Islander people are encouraged to apply.

Start: Flexible.

Demo job created on 11jobs for testing and demonstrations. It is not an actual opening at Juneberry Commerce.