Machine Learning Engineer I

Content moderation runs around the clock, with reviewers in multiple time zones and languages. Advertising, subscriptions and transaction fees each bring in a significant share of revenue. Product reviews happen on Thursdays, with demos of whatever shipped that week.

The position:

Bulkhead Apps has an opening for a Machine Learning Engineer I. This hire will help build out our ML platform: feature store, training pipelines, model registry and deployment. The ML Platform team trains and serves models that run on every customer request. We need someone who cares about latency as much as accuracy. This role is fully remote within the US.

Main duties:

• Develop RAG pipelines with embeddings, vector stores (pgvector, Pinecone, Weaviate) and rerankers
• Build training pipelines in PyTorch or TensorFlow and run them on GPUs in AWS, GCP or Azure
• Monitor models for drift, data quality issues and latency regressions
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow
• Write clean, tested Python and review teammates' code
• Document model cards, data sources and known failure modes
• Track experiments, artifacts and versions in MLflow or Weights & Biases
• Handle on-call for the ML services the team owns

What you need:

• 1-2 years of software engineering experience, with at least two spent on ML systems in production
• Strong Python and solid engineering habits (testing, packaging, code review)
• Hands-on experience with PyTorch, TensorFlow or JAX
• Knowledge of LLM APIs, prompt design and evaluation methods
• Comfortable with SQL and large-scale data processing in Spark or Ray

Preferred qualifications:

• Guardrails, red teaming or safety evaluation for LLM features
• CUDA or Triton kernel experience

Salary range: $128,000 - $148,000 a year

What you get:

• Health, dental and vision coverage starting on day one
• 401(k) with 4% match
• Paid time off
• Profit sharing
• Referral program
• Employee assistance program

Most hires start within two to three weeks of an offer.

Start: Immediately.

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