Sr. Staff ML Researcher, LLM Algorithmic Optimization
d-Matrix · Santa Clara
InfrastructureremoteSenior Staff$175k–$265k
At a glance
Highlights
- Invent new methods for efficient LLM inference on novel DNN accelerators.
- Tackle precision, compression, and memory challenges for generative AI workloads.
- Help build memory-centric architecture enabling faster, energy-efficient AI.
What they offer
- Equity
- Bonus
Ideal candidate
Applied mathematician at heart who loves turning theory into working inference systems. Driven by hard problems, rigorous experimentation, and collaborating across hardware and software to ship real impact.
Requirements
- Master's or PhD in Computer Science, Mathematics, Statistics, Physics, Electrical Engineering or related field with 5+ years experience
- Strong foundation in machine learning and applied mathematics applied to practical ML problems
- Hands-on experience developing and evaluating ML algorithms, models or research prototypes
- Strong Python programming skills with object-oriented design
- Experience with PyTorch, JAX, or TensorFlow
- Experience improving ML model performance, efficiency or quality via algorithmic or numerical techniques
- Ability to design rigorous experiments and communicate findings to cross-functional partners
Lab profile
About d-Matrix
Pioneers memory-centric compute-in-memory chips and software for faster, energy-efficient AI inference.
What they build
- Digital in-memory compute inference chips
- Rack-scale AI inference systems
- AI inference software stack
Known for
- Compute-in-memory architecture
- Energy-efficient inference design
- Full-stack chips-to-systems expertise
Key facts
- Founded in 2019
- Headquartered in Santa Clara, California
- Backed by Temasek, Playground Global, and M12
Posted · Last seen