wagey.ggwagey.gg
38,923  jobs38,923  jobs
Browse Tech JobsCompaniesFeaturesPricingFAQs
Log InGet Started Free
Jobs(38,923)/AI Engineer Role(1,038)/Axle (2) - AI Engineer II
Axle

Axle - AI Engineer II

Hybrid - USA *$120k - $150k2mo ago
In OfficeMidNACloud ComputingArtificial IntelligenceAI EngineerPythonDocumentationDockerKubernetesLinux

Requirements

• 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. • 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. • 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. • Disclaimer: • 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: [email protected]

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.

Benefits

• 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)

Apply in one click

Upload My Resume

Drop here or click to browse · Tap to choose · PDF, DOCX, DOC, RTF, TXT

Apply in One Click
Apply in One Click

Similar roles

Arize AIArize AI - Forward Deployed AI Engineer, West1w ago
·Remote - PT (Pacific)·$125k - $175k/year + Equity
RemoteNAMidCloud ComputingArtificial IntelligenceAI EngineerJavaTypeScriptPythonMLOpsAWSGCPAzureDockerKubernetes
ElasticElastic - AI Engineer2d ago
·Unknown - USA *·$94k - $94k/year
In OfficeNAJuniorCloud ComputingArtificial IntelligenceAI EngineerElasticsearchElastic StackSalesforcePythonTypeScriptKubernetesTerraformDockerDocumentationGCPAWSAzure
Defense UnicornsDefense Unicorns - Forward Deployed AI Engineer1mo ago
·Remote - United States·$149k - $149k/year + Equity
RemoteNAMidArtificial IntelligencePublic SectorAI EngineerPythonTypeScriptKubernetesFull StackNoSQL
Zeely Inc.Zeely Inc. - AI Engineer (Image/Video Generation)2mo ago
·Remote - USA
RemoteNAMidArtificial IntelligenceAI EngineerProduct MarketingMarket ResearchContent CreationPythonDocker
ZapierZapier - Sr. Applied AI Engineer2mo ago
·Remote - San Francisco, California, United States·$232k - $348k/year
RemoteNASeniorCloud ComputingArtificial IntelligenceAI EngineerTypeScriptPythonDocumentationCloseLater
1mind1mind - AI System Engineer1mo ago
·Remote - USA·$175k - $250k/year
RemoteNAArtificial IntelligenceAI EngineerPythonKubernetesTypeScript
DatacorDatacor - AI Optimization Engineer2mo ago
·Remote - USA
RemoteNAMidCloud ComputingArtificial IntelligenceAI EngineerC++JavaPythonAWS
CoreStoryCoreStory - AI Engineer5mo ago
·Remote US - Hybrid·Equity
In OfficeNAMidCloud ComputingSoftwareArtificial IntelligenceAI EngineerPythonFastAPIPandasMilvusPinecone
fastino-aifastino-ai - AI Engineer4mo ago
·Remote - United States
RemoteNAMidArtificial IntelligenceAI EngineerKubernetesVector

Browse more by category

Show 1,038 moreAI EngineerShow 6,296 morePythonShow 5,758 moreDocumentationShow 1,082 moreDockerShow 1,913 moreKubernetesShow 979 moreLinux
Privacy·Terms··Contact·FAQ·Wagey on X