AI Research Engineer, Inference

Requirements

  • Strong engineering skills, especially any of: CUDA/Triton/Pallas/CuTe DSL kernel development, lower-level PyTorch/JAX/XLA development, CUDA Graphs, FPGA/ASIC experience
  • Must have two or more years work experience building deep learning systems, for any domain: robotics, biology, chemistry, physics, audio, video, recommendations, etc.
  • Experience translating methods between areas of application is highly valued
  • LLM experience is valuable, but not necessary
  • Finance experience is not required

Perks and Benefits

  • Base salary range: 250,000 to 300,000 USD per year
  • Eligible for discretionary performance-based bonuses
  • Competitive benefits package
AI Summary ✨

Similar jobs

  • Machine Learning Engineer

    MetaLondon, UK2 days ago
    NewRegistered sponsor
  • Senior Staff Machine Learning Engineer

    MetaLondon, UK2 days ago
    NewRegistered sponsor
  • Research Engineer, Gemini Large Scale Pretraining, DeepMind

    GoogleLondon, UK3 days ago
    Registered sponsor
  • See all jobs in UK →