Member of Technical Staff, Captions
Image Generation
$175-275k
+ Offers Equity
More information about location
Office located in Union Square, NY
AI-powered creative studio
Posted today
AI-powered creative studio
21-100 employees
Posted today
$175-275k
+ Offers Equity
More information about location
Office located in Union Square, NY
21-100 employees
Company mission
To empower our customers, regardless of background or experience, with the ability to bring their vision to life in seconds.
Role
Who you are
- Master’s or PhD in Computer Science, Machine Learning, or a related field or equivalent practical experience
- Demonstrated experience implementing and improving state‑of‑the‑art generative image models
- Deep expertise in generative modeling approaches (flow matching / diffusion, autoregressive models, VAEs, GANs, etc.)
- Strong background in optimization techniques, sampling, and loss‑function design
- Experience with empirical scaling studies and systematic architecture research
- Track record of research contributions at top ML conferences (NeurIPS, CVPR, ICCV, ICML, ICLR)
- Strong proficiency in modern deep‑learning tooling (PyTorch, CUDA, Triton, FSDP, etc.)
- Experience training image diffusion models with billions of parameters
- Familiarity with large language models or multimodal transformers is a plus
- Deep understanding of attention, transformers, latent representations, and modern image‑text alignment techniques
- Expertise in distributed training systems, model parallelism, and high‑throughput inference
- Proven ability to implement and improve complex model architectures end to end
- Ability to write clean, modular research code that scales from prototype to production
- Strong software‑engineering practices including testing, code review, and CI/CD
- Experience with rapid prototyping and experimental design under tight iteration loops
- Strong analytical skills for debugging model behavior, numerical stability, and performance bottlenecks
- Familiarity with profiling and optimization tools (Nsight, TensorBoard, PyTorch Profiler, etc.)
- Track record of bringing research ideas to production and maintaining high code quality in a research environment
What the job involves
- Captions is seeking an exceptional Member of Technical Staff to advance the state‑of‑the‑art in large‑scale image generation models
- You’ll conduct novel research on generative image models for people and storytelling, developing new training techniques and scaling models to billions of parameters and millions of users
- As a key member of our AI team, you’ll work at the cutting edge of image generation systems that enable natural, expressive, and high‑fidelity outputs
- Our team has strong expertise in training large‑scale models with demonstrated research and product impact (see our recent whitepaper here for details of recent work)
- We’re especially excited to push the use of image synthesis for multimodal video generation, with a focus on photorealistic quality, professional-grade expressivity, and creative iteration
- Our models power tools used by millions of creators, and we’re tackling fundamental challenges in how to generate compelling composition, lighting, and fine‑grained detail across diverse domains
- Design and implement large‑scale image generation models (transformers, latent diffusion, flow matching, etc.)
- Develop new approaches to multimodal conditioning and generation (e.g. audio and video) and controllability (editing, multi-frame consistency, script guidance, etc)
- Research advanced image‑editing and -generation techniques such as content‑preserving edits, multi‑input conditioning, and reference‑based generation
- Establish and validate scaling laws for image diffusion models across resolution and parameter count
- Develop automated evaluation approaches for improved fidelity and consistency
- Drive rapid experimentation with model architectures, sampling strategies, and training strategies
- Validate research directly through product deployment and real user feedback
- Derive insights from data and recommend architectures and training practices that will make meaningful impacts on our products
- Train and optimize models at massive scale (10s–100s of billions of parameters) across multi‑node GPU clusters
- Push the boundaries of efficiency and hardware utilization for training and deploying models in a cost effective manner
- Develop sophisticated distributed training approaches using FSDP, DeepSpeed, Megatron‑LM, Triton and custom CUDA kernels where needed
- Design and implement model‑compression techniques (pruning, distillation, quantization, etc.) for efficient serving
- Create new approaches to memory optimization, gradient checkpointing, and mixed‑precision training
- Research techniques for improving sampling speed (DDIM, PFGM++, SDE‑VE) and training stability at scale
- Conduct systematic empirical studies to benchmark architecture and optimization choices
- You’ll work directly alongside our research and engineering teams in our NYC office
- We’ve intentionally built a culture where technical innovation and research excellence are highly valued
- Your success will be measured by your contributions to improving our models and advancing the field, not by your ability to navigate politics
- We’re a team that loves diving deep into complex technical problems and emerging with practical breakthroughs
Our take
As AI integration grows in creative fields, concerns arise about the future of professions like writing and design. However, Captions has transformed these anxieties into opportunities by harnessing AI technology to streamline the creative workflow.
By automating tasks in music, graphic design, and video production, Captions allows users to generate compelling content directly from their mobile devices. With a focus on meeting the rising demand for captivating digital content, its subscription-based model covers key processes like captioning, editing, dubbing, and soundtracking.
Captions' innovative approach has garnered attention, securing funding from tech giants like Sequoia Capital and Andreessen Horowitz. This investment will grow the company's machine learning team and invest further into in-house research efforts and technical infrastructure.
Freddie
Company Specialist at Welcome to the Jungle
Company
Funding (2 rounds)
Jul 2024
$60m
SERIES C
Jun 2023
$25m
SERIES B
Total funding: $85m
Company benefits
- Comprehensive medical, dental, and vision plans
- Anything you need to do your best work
- We’ve done team off-sites to places like Paris, London, Park City, Los Angeles, and Nashville with more planned in the future
Company HQ
Civic Center, New York, NY
Leadership
Gaurav Misra
(CEO)Previously the Head of Design Engineering at Snap Inc, Senior Software Engineer at Localytics, Software Development Engineer at Microsoft, and Software Engineer at Lattice Engines.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 14 more jobs at Captions