AI Researcher (Intern)

Gimlet Labs, Inc; San Francisco (CA); On-site; USD 30,000 - 50,000; Full time

Job summary

Gimlet Labs, Inc in San Francisco is looking for an AI Research Intern to evaluate and implement performance optimizations for AI models. You will work on cutting-edge AI research projects, exploring new model architectures and experimenting with advanced inference techniques.

The ideal candidate is pursuing a degree in computer science or engineering and has experience with AI/ML and distributed systems. Join us to shape the future of AI infrastructure in a dynamic team environment.

Qualifications

• Currently pursuing a degree in computer science, engineering, or comparable area of study.
• Experience with AI/ML or distributed systems.

Responsibilities

• Monitoring and evaluating cutting-edge AI research.
• Researching ways to improve model accuracy, performance, and efficiency.
• Prototyping frameworks with the latest fine-tuning and distillation techniques.

Skills

AI/ML experience

Distributed systems experience

Familiarity with PyTorch

Familiarity with TensorFlow

Software development with Python

Software development with C++

Education

Degree in computer science or engineering

Job description

About Us

Gimlet is building the next generation of AI infrastructure: large-scale AI datacenters and the orchestration platform that coordinates them.

The future of AI will require vastly more compute than exists today. But as AI workloads become more complex and new hardware architectures emerge, simply deploying more GPUs isn't enough. The challenge is making increasingly diverse compute work together.

Gimlet's platform intelligently partitions and routes workloads across heterogeneous hardware, enabling step-function improvements in performance and efficiency. Customers deploy through production-grade APIs without needing to think about hardware selection, placement, or optimization.

We work with foundation labs, hyperscalers, and AI-native companies to power production workloads at massive scale and help define the infrastructure layer for the future of AI.

About the role

Gimlet Labs is seeking an Member of Staff focused on AI Research (Intern). As an AI Researcher (Intern), you will be evaluating and implementing techniques to drive performance and quality optimizations across the latest AI models. The research team is responsible for exploring new model architectures and experimenting with novel inference efficiency techniques such as KV caching and FlashAttention. The team will design and prototype frameworks leveraging fine-tuning and knowledge distillation to push the boundaries of model performance.

What you will work on

• Monitoring and evaluating cutting-edge AI research
• Researching ways to improve model accuracy, performance and efficiency
• Prototyping frameworks with the latest fine-tuning and distillation techniques

You may be a good fit if

• Currently pursuing degree in computer science, engineering, or comparable area of study
• Experience with AI/ML or distributed systems.

Strong candidates may also have

• Experience with PyTorch, TensorFlow, ONNX and other AI frameworks
• Familiarity with distributed systems and orchestration frameworks (e.g., Kubernetes)
• Software development experience with Python and C++
• Understanding of the latest AI research and techniques

What Makes Gimlet Different

At Gimlet, you will work on infrastructure problems that span the full stack of modern AI systems. Our team operates across datacenters, networking, distributed systems, compilers, runtimes, orchestration, and performance engineering to build the foundation for the next generation of AI infrastructure.

As an early member of the team, you will have significant ownership, work alongside highly technical engineers, and help shape both the systems we build and how we scale the company.

We value people who are excited to work across domains, take ownership of meaningful problems, and build technology that enables the next generation of AI.