Senior Software Engineer, Machine Learning
Core
Building production-grade transformer models and scalable inference systems to help legal professionals manage unstructured information and extract insights.
Role type
Senior IC machine learning engineer (LLMs & inference)
Builds
Production ML models, simulation environments, and scalable inference systems for legal tech
Domain
Legal technology + Large Language Models (LLMs)
Deliverable
production ML models
Required skills
Transformer model training, LLM pipeline iteration, PyTorch, inference engines (vLLM, SGLang, TensorRT), weight quantization, MLA/GQA nuances
Preferred skills
Synthetic data pipeline design, Gym-style simulation environments, experimental ML initiative leadership
Technologies
PyTorch, vLLM, SGLang, TensorRT
Responsibilities
Fine-tune and iterate on transformer-based models; Build scalable, low-latency inference systems with monitoring and dynamic scaling; Design synthetic data pipelines and simulation environments; Lead experimental ML initiatives; Shape foundational systems for Legal AGI
Seniority
Senior, hands-on IC