We are seeking a Lead Engineer / Full stack developer to join a legal technology AI start up, The role will work alongside the CTO to focus on building the firm's core platform for law firm automation.
In this early stage, well funded start up, you will work at the cutting edge of modern ML and LLM techniques. Python, and ML framework Py torch, Tensor Flow or Hugging Face
About the Role:
* Design and implement document intelligence pipelines for contract parsing, data extraction, and document classification
* Build and fine-tune LLM-based systems for legal document analysis, clause identification, and obligation extraction
* Develop NLP models for entity recognition, relationship extraction, and semantic understanding of legal text
* Create and curate training datasets from Australian legal documents (conveyancing contracts, leases, facility agreements)
* Integrate AI capabilities into the company's product suite via well-designed, reliable APIs
* Design evaluation frameworks to benchmark model performance against legal expert baselines
* Collaborate with conveyancing and legal domain experts to validate AI outputs for accuracy and trustworthiness
* Stay current with advances in document AI, transformer architectures, retrieval-augmented generation, and fine-tuning
About You:
* 3+ years of experience in machine learning, NLP, or AI engineering with production deployment experience
* Strong proficiency in Python and ML frameworks (PyTorch, TensorFlow, or Hugging Face Transformers)
* Experience building production ML systems including model training, evaluation, deployment, and monitoring
* Hands-on experience with LLMs — prompt engineering, fine-tuning, RAG architectures, and evaluation
* Understanding of document processing, OCR, and information extraction techniques
* Experience with cloud ML infrastructure (AWS SageMaker, or equivalent)
* Strong software engineering fundamentals — clean code, testing, version control, CI/CD
* Based in or willing to relocate to Sydney or Melbourne, Australia
#J-18808-Ljbffr