Software Engineer - USA

Cogniify, Inc.; Santa Clara, Northern (CA, KY); Hybrid; USD 150,000 - 170,000; Full time

Benefits offered by this job

Unlimited PTO

Parental leave (generous)

Medical insurance

ESPP

Team building

Mentorship

Job summary

Cogniify, Inc. seeks a Senior AI/ML Engineer to lead the design and delivery of production-grade ML systems and drive the maturity of ML engineering and MLOps capabilities.

You will own critical ML workstreams end to end and mentor engineers while collaborating with cross-functional teams to ensure scalable, reliable solutions aligned with business goals. The role covers NLP, CV, recommendation, and forecasting, including end-to-end lifecycle management, orchestration, and governance.

Qualifications

• 8-10 years of professional ML engineering experience with production delivery.
• Deep expertise in Python and ML frameworks (TensorFlow, PyTorch, JAX).
• Experience designing and operating production ML pipelines at scale.
• Strong knowledge of MLOps and tools (MLflow, Kubeflow, Airflow, Argo).
• Experience with cloud-native ML platforms (AWS SageMaker, Azure ML, GCP Vertex AI).
• Experience with model serving at scale (TensorFlow Serving, Triton, BentoML, Seldon).
• Strong understanding of distributed computing, data engineering, and observability.

Responsibilities

• Lead design, development, and production deployment of ML systems across NLP, CV, recommender, and forecasting domains.
• Own end-to-end ML lifecycle from problem framing to deployment and monitoring.
• Architect scalable, fault-tolerant ML pipelines with modern orchestration and serving frameworks.
• Drive MLOps adoption including CI/CD, automated retraining, model registry and governance.
• Define engineering standards for model development, testing, and documentation.
• Mentor mid-level and junior engineers; conduct design and code reviews.

Skills

Python

TensorFlow

PyTorch

JAX

MLflow

Kubeflow

Airflow

Argo Workflows

AWS SageMaker

Azure ML

GCP Vertex AI

TensorFlow Serving

Triton

Seldon

Education

Bachelor's degree in CS/math/statistics

Master's degree in related field

PhD (plus)

Tools

MLflow

Kubeflow

Airflow

Argo Workflows

SageMaker

Vertex AI

GCP Vertex AI

Job description

About the Role

We are looking for a Senior AI/ML Engineer to lead the design and delivery of production-grade machine learning systems and drive the maturity of our ML engineering and MLOps capabilities. In this role, you will own critical ML workstreams end to end, make key architectural decisions, mentor other engineers, and collaborate closely with cross-functional teams to ensure ML solutions are scalable, reliable, and aligned with business objectives. The ideal candidate brings deep technical expertise, proven production experience, and the ability to influence technical direction across the team.

Key Responsibilities

Lead the design, development, and production deployment of complex machine learning systems across multiple domains (NLP, computer vision, recommendation, forecasting, etc.).

Own the end-to-end ML lifecycle from problem framing and data strategy through model development, validation, deployment, and monitoring.

Architect scalable, fault-tolerant ML pipelines using modern orchestration and serving frameworks.

Drive the adoption and maturity of MLOps practices including CI/CD for ML, automated retraining, model registry, and governance.

Define and enforce engineering standards for model development, testing, code quality, and documentation.

Evaluate and introduce new tools, frameworks, and techniques to improve model performance, pipeline efficiency, and developer productivity.

Collaborate with data engineers, platform engineers, and product teams to align ML infrastructure with organizational goals.

Mentor and provide technical guidance to mid-level and junior engineers.

Conduct design reviews, code reviews, and architectural assessments for ML systems.

Contribute to technical roadmap planning and communicate tradeoffs and recommendations to engineering leadership.

Identify and mitigate risks related to data quality, model drift, bias, and security in production ML systems.

Required Qualifications

Bachelor’s or Master’s degree in Computer Science, Mathematics, Statistics, or a related field. PhD is a plus.

8-10 years of professional experience in ML engineering, applied ML research, or a closely related role with significant production delivery.

Deep expertise in Python and advanced proficiency with ML frameworks such as TensorFlow, PyTorch, or JAX.

Extensive experience designing and operating production ML pipelines at scale.

Strong knowledge of MLOps principles and tools including MLflow, Kubeflow, Airflow, Argo Workflows, or similar.

Proven experience with cloud-native ML platforms (AWS SageMaker, Azure ML, GCP Vertex AI) and infrastructure-as-code practices.

Experience with model serving at scale using frameworks such as TensorFlow Serving, Triton, BentoML, or Seldon.

Strong understanding of distributed computing, data engineering, and scalable system design.

Experience with monitoring, observability, and governance for production ML systems.

Demonstrated ability to mentor engineers and influence technical direction.

Excellent communication skills with the ability to present technical concepts to both technical and non-technical audiences.

Preferred Qualifications

Experience with LLM-based systems, RAG pipelines, or agentic AI architectures.

Experience with feature platforms (Feast, Tecton) and data quality frameworks (Great Expectations, Deequ).

Familiarity with model explainability and fairness tools (SHAP, LIME, Fairlearn).

Experience with real-time ML serving and streaming data pipelines (Kafka, Flink).

Contributions to open-source ML/MLOps projects.

Experience with GPU cluster management and cost optimization for training workloads.

Salary Range

US East/West Coast: $150,000 - $170,000

Disclaimer: The base salary range is a guideline and may vary based on factors such as candidate experience, specialized skills, and geographical location. Actual compensation may include additional benefits and bonuses.

Perks And Benefits Of Working With Us

Unlimited PTO.

Please ask us about our very generous parental leave, much above industry standards!.

Entrepreneurial culture where pushing limits and taking risks is everyday business.

Open communication with management and company leadership.

Small, dynamic teams = massive impact.

Medical, Dental and Vision coverage for employees.

Access to Disability & Life insurance.

Mental health and wellbeing support

Employer Stock Purchase Program (ESPP)

Yearly Team building experiences

Mentorship and sponsorship opportunities

Manager resources and support

We are an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, veteran status, or any other protected characteristic.