Job Details

5 days ago Be among the first 25 applicants

Direct message the job poster from Mumba Technologies, Inc.

We are team of chip design, analog and AI domain experts working towards increasing

Generative AI adoption by leveraging novel in-memory compute hardware to vastly enhance

the power efficiency of model deployment at the edge. We are building something entirely

new to make AI ubiquitous in a responsible way – widely applicable, economically viable,

and environmentally sustainable.

About the Role

The AI Framework Architect is a tech lead role building the highest level of the compiler stack,

where models are ingested from PyTorch/ONNX/Keras and converted to a proprietary

analog graph that can be parsed by the lower levels of the stack for hardware deployment.

The role requires simultaneous tech lead and individual contributor (IC) work hats since the

development team is highly skilled and motivated, but the complexity of the compiler

requires deep expertise in graph, deep learning and software engineering principles to

overcome hurdles presented by the novel chip architecture. You need to write code

shoulder-to-shoulder with your team.

Responsibilities

As the tech lead for the AI framework, you will be responsible for these key facets of model

deployment:

• Compiler Integration: continue to build and integrate Python-based functionality to

ingest, parse and prepare model deployment to hardware

• Analytical Modeling: understand QAT, PTQ and analog-related modeling principles

• Deep Learning: understand at a fundamental level how these models/networks are

defined to train them for our specific workload requirements

• Graph Computation: fully comprehend at an operational level to direct graph

conversion processes at lower levels of the compiler stack

• Team Development: the individual will lead a diverse, highly talented, motivated,

remote team of engineers, and must be invested not only in product deliverables, but

also in team members' growth in terms of their professional and technical skill sets.

Candidate Requirements

The candidate will have:

• Python: 5+ years in Python development experience as an IC

• Tech Lead: have led a team of engineers in developing production-level

deliverables in deep learning/AI

• Software Engineering: understand how to manage production workflows via

Git/BitBucket and how to direct development efforts with Jira tasking

• Compilers: have experience working on production or proprietary compilers

• Deep Learning: have built and trained DL pipelines for models in either computer

vision or NLP domain, using one or multiple DL frameworks such as PyTorch,

• Analytics: have an established background working on quantization techniques,

stochastic processes and/or advanced numerical principles

Ideally, the candidate will have:

• Python: 8+ years in Python development

• Publication Track Record: published papers in the DL/AI in conferences or

journals, and/or holding pertinent patents

• AI Accelerators: have prior work history with other AI accelerator platforms such as

Location & Commitments

We are on a hybrid work schedule, requesting two days a week in office (Santa Clara)

presence with the remaining work week being operated remotely. The hours are challenging

since the development team is global, with members working from Germany, Australia and

Sri Lanka. There is flexibility on the in-office presence requirement, but the tech lead role is

essential for many facets of the company and needs to be readily available for meeting

sessions during PST work hours.

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Job function

  Information Technology

• Industries

  Information Services, Software Development, and Computer Hardware Manufacturing

Referrals increase your chances of interviewing at Mumba Technologies, Inc. by 2x

Sign in to set job alerts for “Deep Learning Engineer” roles.

Mountain View, CA $158,000.00-$200,000.00 1 week ago

Applied Scientist / Machine Learning Engineer, Software Engineer (Machine Learning)

AI Engineer & Researcher - Post-training

Machine Learning Engineer, Large Behavior Models - Data

Machine Learning Engineer, Large Behavior Models - Computer Vision

Palo Alto, CA $180,000.00-$440,000.00 3 weeks ago

Santa Clara, CA $148,909.00-$252,000.00 3 months ago

San Jose, CA $130,000.00-$182,000.00 9 months ago

AI Engineer & Researcher - Pre-training Scaling, Data, and Eval

Palo Alto, CA $180,000.00-$440,000.00 3 weeks ago

San Jose, CA $119,000.00-$177,000.00 2 weeks ago

Palo Alto, CA $100,000.00-$160,000.00 1 month ago

Santa Clara, CA $148,909.00-$252,000.00 3 months ago

Sunnyvale, CA $80,000.00-$85,000.00 3 hours ago

Palo Alto, CA $100,000.00-$250,000.00 2 weeks ago

Santa Clara, CA $148,909.00-$252,000.00 3 months ago

Palo Alto, CA $168,000.00-$242,500.00 1 hour ago

San Jose, CA $120,700.00-$228,600.00 3 weeks ago

San Jose, CA $137,500.00-$236,500.00 3 weeks ago

Sunnyvale, CA $167,000.00-$185,500.00 1 week ago

San Jose, CA $120,700.00-$228,600.00 3 weeks ago

Software Engineer, AI Platform - New Grad

San Jose, CA $150,000.00-$220,000.00 11 months ago

Mountain View, CA $240,000.00-$280,000.00 15 hours ago

We're unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.