Back to jobs
New

AI Engineer II

Remote

(ID: 2026-1850)

 

Axle is a bioscience and information technology company that offers advancements in translational research, biomedical informatics, and data science applications to research centers and healthcare organizations nationally and abroad. With experts in biomedical science, software engineering, and program management, we focus on developing and applying research tools and techniques to empower decision-making and accelerate research discoveries. We work with some of the top research organizations and facilities in the country including multiple institutes at the National Institutes of Health (NIH).

 

Benefits We Offer:

  • 100% Medical, Dental & Vision Coverage for Employees
  • Paid Time Off and Paid Holidays
  • 401K match up to 5%
  • Educational Benefits for Career Growth
  • Employee Referral Bonus
  • Flexible Spending Accounts:
    • Healthcare (FSA)
    • Parking Reimbursement Account (PRK)
    • Dependent Care Assistant Program (DCAP)
    • Transportation Reimbursement Account (TRN)

 

Role Summary 

Axle is seeking an AI Engineer II with experience designing and delivering production AI systems to lead the development of intelligent applications that support translational research and scientific operations. This role involves architecting retrieval-augmented generation (RAG) systems, agentic pipelines, document understanding workflows, and the shared platform services that make these systems reliable and reusable. 

 

A significant portion of this role involves working directly with clients — scientists, program managers, and operational staff — who understand the problem domain but may not have a technical background. You will partner with them to elicit requirements, translate domain needs into well-defined technical solutions, and ensure delivered software meets their real-world needs. The role requires a self-directed engineer who can make sound architectural decisions, take ownership of technical components end-to-end, mentor junior team members, and deliver iteratively while managing client relationships and expectations. 

 

Key Responsibilities 

  • Architect, develop, and maintain AI-powered applications in Python, including conversational assistants, agentic tools, and retrieval-augmented generation (RAG) systems.
  • Design and optimize retrieval pipelines: embedding generation, vector search, hybrid search, and reranking against large domain-specific corpora.
  • Build document ingestion and parsing pipelines that extract structured content (text, tables, figures, citations) from scientific and technical documents for downstream use by AI systems.
  • Integrate structured knowledge — ontologies, controlled vocabularies, and knowledge graphs — with unstructured text to produce evidence-grounded, provenance-aware AI outputs.
  • Develop and operate shared platform services that accelerate delivery of new AI applications: model gateways, agent-tool integrations, authentication, API key management, and observability.
  • Lead technical design decisions and establish best practices for code quality, testing, evaluation, and deployment of AI systems.
  • Provision, deploy, monitor, and troubleshoot services on Kubernetes and cloud infrastructure, including container builds, manifests, and CI/CD pipelines.
  • Design and maintain relational and vector databases, including schema design, indexing, migrations, and incremental data updates.
  • Define and track quality metrics for AI systems (retrieval quality, answer faithfulness, latency, cost) and drive measurable improvements over time.
  • Engage directly with non-technical clients to gather requirements, clarify goals, manage expectations, and validate that delivered solutions align with their objectives.
  • Translate high-level client and mission needs into actionable technical specifications and development plans.
  • Mentor junior engineers and contribute to a culture of technical excellence.
  • Participate in and help facilitate standups, sprint planning, and cross-team review meetings across multiple project streams. 

 

Required Skills & Qualifications 

  • 5+ years of professional experience in software development, with recent hands-on work building AI or ML-powered applications. 
  • Strong proficiency in Python for application development, scripting, and tooling. 
  • Demonstrated experience building retrieval-augmented generation (RAG) systems, including embeddings, vector search, and reranking. 
  • Experience integrating large language models (LLMs) into production applications using modern agent or orchestration frameworks. 
  • Experience with vector databases and semantic search. 
  • Demonstrated experience making architectural decisions and owning technical direction for software projects. 
  • Experience with Docker, containerization workflows, and Kubernetes for deploying and operating production services. 
  • Experience with PostgreSQL and relational database design. 
  • Strong command of Linux development environments. 
  • Experience with Git/GitHub workflows, including branching, pull requests, and code review practices. 
  • Proven ability to work independently, prioritize across concurrent workstreams, and drive projects to completion with minimal oversight. 

 

Preferred Qualifications 

  • Experience with document parsing, OCR, and layout-aware extraction for scientific or technical documents. 
  • Familiarity with knowledge graphs, ontologies, and controlled vocabularies, including RDF, SPARQL, or graph query languages. 
  • Experience with agent-tool integration frameworks, including the Model Context Protocol (MCP). 
  • Experience with LLM observability, tracing, and evaluation tooling. 
  • Experience with GPU computing (CUDA/NVIDIA) for embedding or inference workloads. 
  • Experience with AWS (IAM, S3, deployment, general infrastructure management). 
  • Familiarity with OIDC/OAuth authentication and API key management in production systems. 
  • Background in biomedical informatics, translational research, or health sciences. 
  • Experience with AI-assisted development tools (e.g., Claude Code, GitHub Copilot). 
  • Familiarity with Jira or similar project management tools. 

 

Interpersonal Skills 

  • Strong written and verbal communication skills, including the ability to convey technical concepts to non-technical collaborators and translate domain expertise into software requirements. 
  • Skilled at requirements elicitation — able to ask the right questions, listen actively, and bridge the gap between what subject matter experts need and what engineering can deliver. 
  • Ability to work effectively across multiple teams and project streams with different leads, priorities, and cadences. 
  • Comfortable operating in a large, cross-functional team environment where coordination and alignment are essential. 
  • Proactive in communicating progress, raising blockers, and seeking input from teammates and stakeholders. 
  • Tactful and professional when engaging with clients and external partners; able to navigate sensitive situations and escalate appropriately. 
  • Receptive to feedback and committed to iterative improvement. 

 

Disclaimer: The above description is meant to illustrate the general nature of work and level of effort being performed by individuals assigned to this position or job description. This is not restricted as a complete list of all skills, responsibilities, duties, and/or assignments required. Individuals may be required to perform duties outside of their position, job description or responsibilities as needed.

The diversity of Axle’s employees is a tremendous asset. We are firmly committed to providing equal opportunity in all aspects of employment and will not tolerate any illegal discrimination or harassment based on age, race, gender, religion, national origin, disability, marital status, covered veteran status, sexual orientation, status with respect to public assistance, and other characteristics protected under state, federal, or local law and to deter those who aid, abet, or induce discrimination or coerce others to discriminate.

Accessibility: If you need an accommodation as part of the employment process please contact: careers@axleinfo.com

This role has a market-competitive salary with an anticipated base compensation range listed below. Actual salaries will vary depending on a candidate’s experience, qualifications, skills, and location.

#IND

 

Salary Range

$120,000 - $150,000 USD

Create a Job Alert

Interested in building your career at Axle? Get future opportunities sent straight to your email.

Apply for this job

*

indicates a required field

Resume/CV*

Accepted file types: pdf, doc, docx, txt, rtf

Select...
Select...
Select...
Select...
Select...
Select...
Select...

Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.

As set forth in Axle’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

Select...
Select...
Race & Ethnicity Definitions

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

Select...

Voluntary Self-Identification of Disability

Form CC-305
Page 1 of 1
OMB Control Number 1250-0005
Expires 04/30/2026

Why are you being asked to complete this form?

We are a federal contractor or subcontractor. The law requires us to provide equal employment opportunity to qualified people with disabilities. We have a goal of having at least 7% of our workers as people with disabilities. The law says we must measure our progress towards this goal. To do this, we must ask applicants and employees if they have a disability or have ever had one. People can become disabled, so we need to ask this question at least every five years.

Completing this form is voluntary, and we hope that you will choose to do so. Your answer is confidential. No one who makes hiring decisions will see it. Your decision to complete the form and your answer will not harm you in any way. If you want to learn more about the law or this form, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.

How do you know if you have a disability?

A disability is a condition that substantially limits one or more of your “major life activities.” If you have or have ever had such a condition, you are a person with a disability. Disabilities include, but are not limited to:

  • Alcohol or other substance use disorder (not currently using drugs illegally)
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, HIV/AIDS
  • Blind or low vision
  • Cancer (past or present)
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or serious difficulty hearing
  • Diabetes
  • Disfigurement, for example, disfigurement caused by burns, wounds, accidents, or congenital disorders
  • Epilepsy or other seizure disorder
  • Gastrointestinal disorders, for example, Crohn's Disease, irritable bowel syndrome
  • Intellectual or developmental disability
  • Mental health conditions, for example, depression, bipolar disorder, anxiety disorder, schizophrenia, PTSD
  • Missing limbs or partially missing limbs
  • Mobility impairment, benefiting from the use of a wheelchair, scooter, walker, leg brace(s) and/or other supports
  • Nervous system condition, for example, migraine headaches, Parkinson’s disease, multiple sclerosis (MS)
  • Neurodivergence, for example, attention-deficit/hyperactivity disorder (ADHD), autism spectrum disorder, dyslexia, dyspraxia, other learning disabilities
  • Partial or complete paralysis (any cause)
  • Pulmonary or respiratory conditions, for example, tuberculosis, asthma, emphysema
  • Short stature (dwarfism)
  • Traumatic brain injury
Select...

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.