ML Engineer (Internship and Full-time)

About the role

Tilde Research is a moonshot AI lab advancing mechanistic interpretability, new architectures, and pretraining science. We build foundational understanding of models to advance the frontier of intelligence.

Responsibilities

• Optimize inference and training throughput for novel model architectures
• Build and maintain high-performance distributed training infrastructure
• Collaborate with researchers to translate insights into measurable improvements in model performance and understanding

Requirements

• Experience in deep learning or related research areas
• Demonstrated exceptional capability in working on ML infrastructure, including strong open source contributions, thoughtful technical blog posts/work logs, previous experience working with large-scale pre/post-training infrastructure, deep familiarity with Pytorch or Jax, and basic familiarity with Triton/Tilelang/TK etc.
• Communicate clearly and effectively, both verbally and in writing
• Able to design and orchestrate end-to-end ML pipelines
• Able to learn quickly

Qualifications

• Master's degree in Computer Science, Electrical Engineering, or a related field
• Relevant work experience in machine learning or related fields

Skills

• Strong programming skills in Python, C++, or other relevant languages
• Familiarity with deep learning frameworks such as PyTorch or JAX
• Experience with distributed computing and parallel processing
• Excellent problem-solving and analytical skills

Benefits

• Competitive salary commensurate with experience
• Flexible work schedule
• Health insurance benefits
• Professional development opportunities

Pay

• $100,000 - $150,000 annually

Schedule

• Full-time