ML Tech Lead

ED Systems is hiring for the ML Tech Lead role to take models from notebook to production and keep them working once they're there. This role is fully remote within the US. Normal working hours, Monday to Friday. Good written English matters here, since most updates are written.

Essential duties and responsibilities:

• Develop RAG pipelines with embeddings, vector stores (pgvector, Pinecone, Weaviate) and rerankers
• Deploy models behind low-latency APIs with FastAPI, Triton, TorchServe or vLLM
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow
• Monitor models for drift, data quality issues and latency regressions
• Cut inference cost with quantization, batching and caching
• 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
• Fine-tune open-weight models with LoRA when it beats prompting

Requirements:

• 7+ years of software engineering experience, with at least two spent on ML systems in production
• Hands-on experience with PyTorch, TensorFlow or JAX
• Familiarity with Docker, Kubernetes and a cloud ML platform (SageMaker, Vertex AI, Azure ML)
• Understanding of core ML concepts: overfitting, evaluation metrics, bias in training data
• BS or MS in computer science, engineering or a related field
• Comfortable with SQL and large-scale data processing in Spark or Ray

Compensation and benefits:

Salary range: $320,000-$375,000/year

• Health savings account (HSA) with employer contribution
• 401(k) with 4% match
• 11 paid holidays

Who we are:

Product, engineering and support teams work out of offices in New York City, New York and two nearshore delivery centers. More than 9,000 organizations use our HR and payroll suite, from regional hospitals to school districts. Releases go out Tuesday mornings, never on Fridays.

Applicants must be authorized to work in the U.S.

Start: Within 30 days.

Demo job created on 11jobs for testing and demonstrations. It is not an actual opening at ED Systems Inc..