# Member of Technical Staff, Research Engineer

**Company:** [Ataraxis AI](https://hotfix.jobs/companies/ataraxis-ai)
**Location:** New York, NY
**Role:** ML Engineering
**Salary:** $120k – $210k/yr
**Skills:** Python, PyTorch, Machine Learning, Deep Learning, Self-Supervised Learning, Survival Analysis, Multi-Modal Learning, Causal Inference, Model Interpretability, Gpu Optimization, Statistics
**Posted:** 2026-07-03

> Research Engineer implementing novel ML models for clinical AI, focusing on self-supervised learning, survival analysis, multi-modal data, causality, and interpretability to predict patient outcomes in precision medicine. Requires strong Python/PyTorch skills, deep learning experience, and statistics foundations; publications are a plus.

## Job Description

## Responsibilities
- Implement novel machine learning models and methods for self-supervised learning, survival analysis, multi-modal learning, causality and interpretability.
- Translate machine learning and statistics papers into production-ready code.
- Build robust model evaluation frameworks and monitor model performance.
- Develop pipelines for data preprocessing, integration, and quality assurance.
- Maintain high standards of scientific documentation to ensure reproducibility and clarity in both internal reports and external publications.
- Optimize code to run efficiently on GPU clusters, with emphasis on speed and scalability.
- Deploy machine learning models to the cloud in optimized inference pipelines.
- Develop and maintain regression and unit tests to ensure high-quality code.
- Disseminate the results by co-authoring research papers and abstracts.
- Collaborate with a multidisciplinary team of engineers and scientists.

## Qualifications
- BS/MS/PhD degree in computer science, machine learning or statistics.
- Excellent understanding of core machine learning concepts.
- Excellent knowledge of the foundations of statistics, linear algebra and probability.
- Excellent skills in Python and PyTorch.
- Proficiency in data visualization and communicating complex results to both technical and non-technical audiences.
- Excellent understanding of computer architecture, parallel training of AI models, and GPU optimization.
- Experience in deep learning.
- Experience in at least one of {self-supervised learning, survival analysis, multi-modal learning, domain adaptation, causal inference, model interpretability, computational pathology} is a plus.
- Attention to detail and ability to drive tasks to completion.
- Passion for research. Prior publications in A* conferences (e.g. ICML, ICLR, NeurIPS, CVPR) are a plus.

## Similar roles

- [Staff Research Scientist](https://hotfix.jobs/jobs/28a93c6b-3142-4c68-84ef-ea842e34456c) - Carbon - Sunnyvale, CA - $138k – $206k/yr
- [Member of Technical Staff (All Levels)](https://hotfix.jobs/jobs/99a9acbb-e952-4c1a-a57d-c99937c854e8) - Basis - New York, NY - $100k – $300k/yr
- [Senior / Staff Applied Research Software Engineer](https://hotfix.jobs/jobs/06386432-24d8-4914-bc6c-9a9874c695db) - Twilio - Remote - $142k – $252k/yr
- [Member of Technical Staff, Machine Learning](https://hotfix.jobs/jobs/0690a304-2c3d-4cff-aa6b-d101ed35ca25) - Sieve - San Francisco, CA - $150k – $350k/yr
- [Member of Technical Staff](https://hotfix.jobs/jobs/0be719d4-4a8a-484f-ad6e-6b8b60347618) - Modal - New York, NY - $150k – $350k/yr

**Apply:** https://hotfix.jobs/jobs/bf12bd5d-a3f0-4847-8106-d096661c1fdb
**Canonical:** https://hotfix.jobs/jobs/bf12bd5d-a3f0-4847-8106-d096661c1fdb