Junior Machine Learning Engineer

Tax Relief Advocates offers a growth-oriented junior machine learning engineer role in Irvine, onsite. The position centers on designing, building, and refining ML models for document classification, entity extraction, and outcome prediction to support tax automation. It provides mentorship within a collaborative team and hands-on exposure to real-world AI applications in tax automation.

Benefits

• Compensation: Up to $85,000 - $100,000 per year, depending on experience.
• Onsite role in Irvine, CA with mentorship and a clear path for growth within a collaborative team.
• Hands-on experience with the Azure stack, including AI Foundry and Document Intelligence, alongside Python development.
• Opportunity to contribute to transformative AI projects that modernize tax automation.
• Supportive culture designed to launch your ML career through structured guidance and ongoing learning.

Responsibilities

• Collaborate on designing, building, and refining ML models for document classification, entity extraction, and outcome prediction using libraries such as scikit-learn, PyTorch, or TensorFlow.
• Work with internal teams to modernize legacy workflows by integrating AI powered tools.
• Assist in defining and tracking performance metrics like recall, precision, and accuracy to improve model effectiveness and reduce errors in tax form extraction and chatbot outputs.
• Support deployment and monitoring of models with MLflow, FastAPI, or Streamlit, including version control, logging, and overseeing production systems.
• Join a dynamic team and gain hands-on experience with practical AI applications in tax automation.

Requirements

• Recent graduate (0–2 years of experience, including internships) in AI, NLP, computer science, or related fields.
• Foundational Python skills.
• Experience with scikit-learn, PyTorch, or TensorFlow.
• Experience using pandas for document extraction.
• Experience with a PDF extraction library.
• Exposure to token estimation libraries.
• Experience integrating large language model APIs and handling their payloads (for example OpenAI or Azure AI).
• Experience with OCR or document intelligence tools such as Azure Document Intelligence or Google Document AI.
• Interest in working with structured and unstructured data, especially financial or tax documents.
• Strong communication skills to bridge technical and non-technical teams.
• Eager to learn and proactive about solving problems.
• Good documentation habits and a collaborative mindset.

Technologies

• Python
• scikit-learn
• PyTorch
• TensorFlow
• pandas
• PDF extraction library
• token estimation libraries
• OpenAI
• Azure AI
• Azure Document Intelligence
• Google Document AI
• MLflow
• FastAPI
• Streamlit
• Docker
• AI Foundry
• Document Intelligence
• Azure Synapse
• Power BI
• Snowflake
• AWS Lambda
• Azure Functions

Nice to have

• Experience with containerization and building custom Docker images.
• Knowledge of security fundamentals such as role based access control or JWT authentication.
• Interest in regulatory or IRS compliance considerations for AI models.
• Exposure to data platforms like Azure Synapse, Power BI, or Snowflake.
• Experience with serverless architectures (AWS Lambda, Azure Functions) is a plus.