Machine Learning Engineer – AI Architecture Research
Employment Information
Report this job
Job expired or something wrong with this job?
Job Description
Machine Learning Engineer focusing on AI architecture research and prototyping next-generation models. Collaborating with a small team to influence a Series-A company's technical direction.
Responsibilities:
- Research and develop new neural network architectures (e.g. alternatives or extensions to Transformers, recurrent / hybrid models, long-context systems)
- Design and run architecture-level experiments (scaling laws, memory mechanisms, compute trade-offs)
- Prototype models end-to-end — from research code to training-ready implementations
- Collaborate with inference and systems engineers to ensure architectures are deployable and efficient
- Analyze model behavior, failure modes, and inductive biases
- Read, reproduce, and extend cutting-edge research papers
- Contribute to internal research notes, benchmarks, and open-source efforts (where applicable)
Requirements:
- Strong background in machine learning fundamentals and deep learning
- Hands-on experience implementing model architectures from scratch
- Solid understanding of:
- Attention mechanisms, RNNs, state-space models, or hybrid architectures
- Training dynamics, scaling behavior, and optimization
- Memory, latency, and compute constraints at the model level
- Comfortable working in PyTorch or JAX
- Ability to move fluidly between theory, experimentation, and engineering
- Clear communicator who can explain architectural trade-offs
- Nice to Have
- Experience with non-Transformer architectures (RNN variants, SSMs, long-context models)
- Background in research-driven startups or open-source ML projects
- Experience with large-scale training or custom training loops
- Publications, preprints, or notable research contributions
- Familiarity with inference optimization and deployment constraints
Benefits:
- Competitive compensation + meaningful equity
Similar Jobs
Senior AI Engineer – AI Solutions
Senior AI Engineer at Sicredi focusing on AI-driven solutions that enhance business and member experience. Collaborating across teams while leading innovative projects in artificial intelligence.
GTM AI Engineer
GTM AI Engineer at Tiger Data enhancing go-to-market processes through AI-powered workflows and solutions. Building efficiencies and automations within Sales, Marketing, and RevOps teams.
Manager – ML & AI, AI Platform
Manager, Machine Learning & AI leading AI Platform team at Zapier. Focused on enabling AI features and infrastructure while partnering with multiple teams.
Staff AI Engineer, GenAI
AI Engineer architecting and operationalizing AI/ML solutions at Teladoc Health. Collaborating with teams to ensure production-grade reliability for AI-driven products.
Senior Staff Technical Program Manager – AI Platform/OS
Senior Staff Technical Program Manager driving execution of internal platform initiatives at Trase Systems. Supporting AI platform development to ensure reliability and measurable outcomes.
AI Engineer
AI Engineer at Gainwell Technologies designing and developing innovative AI solutions for healthcare applications. Focused on building and deploying advanced AI technologies to improve healthcare outcomes.
Senior AI Engineer
AI Engineer at Gainwell Technologies focusing on AI and Generative AI development for healthcare solutions. Engage in end-to-end solution deployment and integration while leading innovation in the field.
AI Engineer Training Program, No Tech Background Needed
Flatiron School offers a paid apprenticeship program for career switchers into software engineering. Engage in hands-on learning and mentorship while building a portfolio through real projects.
AI Engineer Work-Study Program – Bletchley Fellowship
Apprenticeship role for earning while learning at Flatiron School specializing in AI and data science development. Contributing to software projects with mentorship from experienced developers and leaders.
Machine Learning Engineer, AI
Machine Learning / AI Engineer collaborating with clients and optimizing AI solutions. Role involves deployment and continuous improvement of data pipelines in a fully remote setting.
Applied AI Engineer
Applied AI Engineer responsible for identifying and delivering AI capabilities in a fintech and proptech context at Splitero. Collaborating across teams to automate workflows and enhance product efficiency.
Senior AI Engineer
AI Engineer collaborating with teams to enhance predictive models in healthcare payments. Requires deep learning experience and understanding of healthcare data, preferably based in Wisconsin.
AI Engineer – KI-Spezialist
AI Engineer developing intelligent systems at RLVNT, a leading communication agency. Collaborating on innovative projects in artificial intelligence and machine learning.
Senior AI Developer – Architect
Senior AI Developer leading technical direction for AI initiatives at Health Admins. Building and optimizing AI systems in a dynamic remote environment.
Senior AI Engineer
Senior AI Engineer designing and validating AI-driven solutions for a Top 5 global consulting company. Collaborating to create intelligent AI systems and workflows in a remote setting.
Senior AI Architect
Senior AI Architect responsible for defining AI-enabled solution architectures across the enterprise. Working with a Top 5 global consulting company on cutting-edge AI initiatives in a remote setup.
AI Architect
AI Architect designing and implementing end-to-end architectures for AI/ML solutions at Cambium. Collaborates with engineering and product teams to ensure scalable and compliant AI projects.
AI Engineer, Team Lead
Lead AI Engineer managing a high-output ML team at Gravyty, focusing on cutting-edge AI solutions. Responsible for coding, mentorship, and project delivery in AI initiatives.
AI Developer – AI Innovation Cell
AI Developer designing and implementing AI-powered features at Applaudo, focusing on scalable AI solutions and business impact. Collaborate across design, development, and AI to create transformative products.
GTM AI Engineer
GTM AI Engineer at AZX developing AI-powered infrastructure for strategic market engagement. Building and running systems for CRM, marketing automation, and sales enablement.