Junior Machine Learning Engineer
Job Description
Junior Machine Learning Engineer in Irvine, CA, focused on designing, building, and refining ML models for document classification, entity extraction, and outcome prediction to support tax automation.
Responsibilities
- Assist in designing, building, and fine‑tuning ML models for document classification, entity extraction, and outcome prediction using tools like scikit‑learn, PyTorch, or TensorFlow.
- Support internal departments by transforming legacy workflows through the integration of AI powered tools.
- Evaluate performance by establishing metrics such as recall, precision, and accuracy to assess model effectiveness and reduce errors in tax form extraction and chatbot responses.
- Deploy and monitor models with tools like MLflow, FastAPI, or Streamlit, including versioning, logging, and ongoing system monitoring.
- Learn and grow within a dynamic team, gaining hands‑on experience with real world AI applications in tax automation.
Requirements
- Recent graduate (0–2 years of experience, including internships) in AI, NLP, computer science, or related fields.
- Basic proficiency in Python.
- Familiarity with libraries like scikit‑learn, PyTorch, or TensorFlow.
- Familiarity with pandas for document extraction.
- Familiarity with a PDF extraction library.
- Familiarity with token estimation libraries.
- Experience working with LLM APIs and their payload outputs (e.g., OpenAI, Azure AI).
- Experience with OCR/document intelligence tools (e.g., Azure Document Intelligence, Google Document AI).
- Interest in working with structured/unstructured data, especially financial or tax documents.
- Exceptional communication skills to bridge technical and non‑technical teams.
- Eager to learn, with a proactive approach to problem‑solving.
- Strong documentation habits and a team‑oriented 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
Benefits
- Work on transformative AI projects that revolutionize tax automation.
- Gain hands‑on experience with the Azure stack, including AI Foundry and Document Intelligence, alongside Python.
- Join a collaborative team in Irvine, CA, with mentorship to kickstart your ML career.
- Compensation: Up to $85,000 - $100,000 depending on experience.
Nice to Have
- Solid understanding of containerization, with experience building custom Docker images.
- Familiarity with security fundamentals like role‑based access control or JWT based authentication.
- Interest in regulatory or IRS compliance for AI models.
- Exposure to Azure Synapse, Power BI, or Snowflake.
- Experience with serverless architectures (e.g., AWS Lambda, Azure Functions) is a plus.
Similar Jobs
J