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