Member of Technical Staff, Foundation Models

Department: AI

Location: New York HQ

Compensation: $100K – $300K • Offers Equity

Employment Type: FullTime

About Ataraxis AI

Ataraxis is a clinical AI research lab working at the intersection of multi-modal AI and precision medicine. Our goal is to make disease predictable. To accomplish this, we develop new AI methods that predict patient outcomes and treatment response, and build clinical tools to assist physicians in selecting the most optimal treatments for their patients.

Our AI research lab discovers and develops methods to recognize patterns and predict outcomes across complex, multi-modal clinical data. This spans our causality (Ataraxis™ Tau), foundation model (Falcon and Kestrel for digital pathology), and survival analysis research.

Our first clinical products, such as Ataraxis™ Breast for breast cancer, already help patients get the most appropriate treatment across the best academic institutions and community clinics worldwide.

At Ataraxis, you will have a unique opportunity to shape not only the future of our company, but also the future of healthcare. You will join an exceptional team at the forefront of clinical AI research and deployment. Our advisors include AI pioneers such as our founding advisor, Yann LeCun, and distinguished oncologists from top cancer research institutions, all united by the mission to redefine precision medicine.

Ataraxis has raised over $24 million in funding, including a $20 million Series A led by top venture capital funds such as Thiel Capital/Founders Fund (OpenAI, SpaceX, Palantir), Obvious Ventures (AMI Labs, Inceptive, Radical Numerics, Recursion), and AIX Ventures (Hugging Face, Perplexity).

We are an company with a flat organizational structure, where every team member is empowered to actively contribute. Leadership roles are earned by those who demonstrate initiative and consistently deliver exceptional results. Strong work ethic and the ability to prioritize ruthlessly are essential.

Responsibilities

• Design and implement novel self-supervised learning methods for training foundation models.

• Translate machine learning papers into production-ready code.

• Build robust model evaluation frameworks.

• Disseminate the results by co-authoring research papers and abstracts.

• Collaborate with a multidisciplinary team of engineers and scientists.

• Co-mentor junior members of the team.

Qualifications

• PhD degree in machine learning or statistics.

• Passion for research, attention to detail and ability to drive tasks to completion. Strong preference will be given to candidates with papers in A* conferences (e.g. ICML, ICLR, NeurIPS, CVPR).

• Excellent understanding of core machine learning concepts.

• Excellent knowledge of the foundations of statistics, linear algebra, probability and machine learning.

• Excellent skills in Python and PyTorch.

• Experience in deep learning and self-supervised learning. Experience in survival analysis, multi-modal learning, domain adaptation, causal inference, model interpretability and computational pathology is a bonus.