hero

We invest in people who change the way the world works.

Interested in working with them?
64
companies
886
Jobs
Tell us about your professional DNA and get discovered by the amazing companies in our network.

Software Engineer, LLM Inference Frameworks Engineer

Together AI

Together AI

Software Engineering
San Francisco, CA, USA
Posted on Tuesday, June 25, 2024

What You'll Be Doing

  1. Develop and optimize large language model (LLM) inference frameworks.
  2. Optimize GPU and CUDA performance to create an industry-leading, high-performance LLM inference engine.
  3. Design and implement distributed inference infrastructure for LLM.
  4. Build monitoring and management tools to ensure the reliability and scalability of online inference servers.
  5. Identify and resolve system inefficiencies and bottlenecks to improve overall system performance.
  6. Develop tools to analyze bottlenecks and sources of instability, then design and implement solutions.
  7. Collaborate with product teams to provide solutions that meet their requirements.

Qualifications

  1. Bachelor’s degree in Computer Science, Computer Engineering, or a relevant technical field, or equivalent practical experience. Experience in ML engineering optimization is preferred.
  2. Proficient in C/C++, Python, or Rust, with a strong understanding of algorithms and data structures.
  3. Expertise in GPU high-performance computing optimization using CUDA, with an in-depth understanding of computer architecture. Familiarity with parallel computing optimization, memory access optimization, and low-bit computing.
  4. Understanding of deep learning algorithms and neural network architectures.
  5. Familiarity with TensorRT-LLM, ORCA, VLLM, and similar frameworks.
  6. At least 3 years of experience working in ML infrastructure (e.g., PyTorch, SageMaker, etc.) and a solid understanding of deep learning training frameworks such as PyTorch and TensorFlow.

Preferred Qualifications

  1. Knowledge of LLM models and experience in accelerating LLM model optimization.
  2. Familiarity with Rust programming.

About Together AI

Together AI is a research-driven artificial intelligence company. We believe open and transparent AI systems will drive innovation and create the best outcomes for society, and together we are on a mission to significantly lower the cost of modern AI systems by co-designing software, hardware, algorithms, and models. We have contributed to leading open-source research, models, and datasets to advance the frontier of AI, and our team has been behind technological advancement such as FlashAttention, Hyena, FlexGen, and RedPajama. We invite you to join a passionate group of researchers in our journey in building the next generation AI infrastructure.

Compensation

We offer competitive compensation, startup equity, health insurance, and other benefits, as well as flexibility in terms of remote work. The US base salary range for this full-time position is: $160,000 - $230,000 + equity + benefits. Our salary ranges are determined by location, level and role. Individual compensation will be determined by experience, skills, and job-related knowledge.

Equal Opportunity

Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.

Please see our privacy policy at https://www.together.ai/privacy