Artificial Intelligence/Machine Learning (AI/ML) Engineer

Job Description

Base-2 Solutions is seeking an AI/ML Engineer responsible for building, training, fine-tuning, and evaluation advanced language models. The ideal candidate will have hands-on experience with state-of-the-art machine learning techniques, specifically for natural language processing (NLP), and will also have a strong understanding of full-stack development. This person will work across the model development lifecycle and collaborate with cross-functional teams to deliver high-impact solutions.

Required Skills

• Technical Skills:
  • Strong proficiency in AI/ML frameworks such as TensorFlow, PyTorch, or Hugging Face
  • Expertise in model training, evaluation, and deployment
  • Hands-on experience with AWS tools like SageMaker, Lambda, EC2, and S3.
  • Experience in developing full-stack software applications (JavaScript, Python, Java, etc.)
  • Solid understanding of data structures, algorithms, and system design
• Experience & Background:
  • Extensive experience in machine learning model development, including natural language processing (NLP)
  • Experience with model evaluation, optimization, and performance monitoring
  • Prove experience in software engineering with strong coding skills
  • Experience working in a government or defense-related environment is highly preferred
• Additional Skills:
  • Experience with model versioning and management using tools such as MLFlow or Git
  • Familiarity with containerization tools (i.e. Kubernetes)
  • Strong problem-solving and analytical skills
  • Excellent communication and collaboration abilities

Qualifications

• Two (2) years experience in applied machine learning in programs and contracts of similar scope, type, and complexity is required.
• A Master's or Ph.D. degree in advanced math, artificial intelligence, data science, computer science or deep learning from an accredited college or university.
  • Five (5) additional years of machine learning experience with a relevant Bachelor's degree may be substituted for a Master's degree.
• Experience with standard machine language frameworks, e.g. Pytorch, TensorFlow.

Desired Skills

• Experience with secure data handling and compliance requirements
• Experience working with Large Language Modes (LLMs)
• Knowledge of Computation Linguistics
• Willingness to support occasional on-call duties is a plus

Responsibilities:

• Knowledge and experience of Language models is required.
• Specific experience with Marian an multi-lingual model development.
• Experience with Open-Source model libraries, Deep Learning Containers (DLC), GPU technologies and optimization/tuning.
• Model Development & Training:
  • Build, train, and fine-tune machine learning models, particularly language models
  • Apply best practices in model training, tuning, and optimization
  • Design and implement solutions for model performance improvement
• Evaluation & Testing:
  • Conduct rigorous model evaluation, including performance analysis and benchmarking
  • Perform error analysis, debugging, and model diagnostics to ensure quality and reliability
• Model Deployment & Integration:
  • Work with cloud-based AI platforms (especially AWS Sagemaker) to deploy and scale models
  • Integrate machine learning models into production environments ensuring seamless integration with other systems
• Full-Stack Engineering:
  • Contribute to the development and maintenance of the full stack for AI model-based applications (front-end and back-end)
  • Collaborate with software engineers to build scalable and efficient deployment pipelines.
• Collaboration & Reporting:
  • Work closely with data scientists, product teams, and engineers to translate business requirements into technical solutions
  • Document workflows, model design processes, and technical specifications

Pay & Benefit Highlights

Compensation

• Above market fixed salary or hourly pay.
• Up to $10,000 bonus for each referral.
• Additional bonuses for exceptional performance, assisting with business development and company growth.

Health

• 100% paid premiums for health insurance. Choose from over 80 gold-level medical plans from Aetna, CareFirst, Kaiser and UnitedHealthcare. Choose from PPO, EPO, POS, HMO, and HSA-compatible.
• HSA and FSA options.
• 100% paid premiums for dental insurance.
• 100% paid premiums for vision insurance.

Income Protection

• 100% paid premiums for short-term disability.
• 100% paid premiums for long-term disability.
• 100% paid premiums for accidental death & dismemberment.
• 100% paid premiums for life insurance with a $200,000 max benefit.

Retirement

• 8% company contribution to 401k with immediate vesting.
• 401k pre-tax and Roth options.

Leave

• Up to 20 days of flexible paid time off (PTO).
• 11 days of paid floating holidays.

Work-Life Balance

• Flexible work schedules including flex time and compressed work period.
• Remote work including partial or fully remote (contract and project-dependent).

View our detailed

Equal Opportunity Employer

Base-2 Solutions is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, or status as a protected veteran.