AI Engineer

Work set-up: 
Full Remote
Contract: 
Experience: 
Junior (1-2 years)
Work from: 

Offer summary

Qualifications:

6-24 months of hands-on Python development experience, preferably with FastAPI or Flask., Working knowledge of at least one ML framework such as TensorFlow, PyTorch, or Keras., Solid understanding of SQL and basic data-wrangling using NumPy and Pandas., Bachelor’s degree in Computer Science, Data Science, or a related field..

Key responsibilities:

  • Containerize scikit-learn/TensorFlow models as FastAPI or gRPC endpoints with low latency.
  • Build reproducible ETL pipelines using Pandas or PySpark and orchestrate with Prefect or Airflow.
  • Author unit and integration tests, and integrate GitHub Actions for CI/CD processes.
  • Collaborate with Data Engineers and Product teams on data contracts and KPI definitions.

Shift Boolean logo
Shift Boolean
11 - 50 Employees
See all jobs

Job description

This is a remote position.


ShiftBoolean is seeking a hands‑on Junior ML Engineer I to turn cutting‑edge research notebooks into production‑grade services. You will convert prototypes into FastAPI micro‑services, refine feature pipelines, and help ship real‑time lead‑scoring and generative‑AI chatbots that power our SaaS products. This is a launch‑pad position aimed at graduates with 6 – 24 months of Python experience who are eager to own deliverables from sprint one.
 Key Responsibilities

  • Model Operationalisation — containerise scikit‑learn / TensorFlow models as FastAPI or gRPC endpoints with < 250 ms P95 latency.

  • Feature Engineering Pipelines — build reproducible ETL in Pandas / PySpark; orchestrate with Prefect or Airflow.

  • Testing & CI/CD — author unit + integration tests; integrate GitHub Actions for automated build, lint, and security scan.

  • Monitoring & Observability — instrument OpenTelemetry metrics; set up Datadog dashboards and drift alerts.

  • Code Review & Documentation — produce clear PRs, RFCs, and internal wiki pages; uphold ShiftBoolean’s coding standards.

  • Cross‑functional Collaboration — pair with Data Engineers on data contracts and with Product on KPI definition.



Requirements
Mandatory Qualifications
  • 6 – 24 months hands‑on Python development (FastAPI or Flask).

  • Working knowledge of one ML framework (TensorFlow, PyTorch, or Keras).

  • Solid grasp of SQL (window functions, CTEs) and basic data‑wrangling (NumPy, Pandas).

  • Experience with Git, RESTful API design, and Docker.

  • Bachelor’s degree in Computer Science, Data Science, or related field.

 Preferred/Bonus Skills
  • Exposure to Spark, Dask, or Ray for large‑scale data processing.

  • Familiarity with experiment‑tracking tools (MLflow, Weights & Biases).

  • Basic understanding of Kubernetes and Helm.

  • Knowledge of cloud services (Azure/AWS) and IaC (Terraform).

​ Soft Skills & Culture Fit
  • Growth mindset, hunger to learn and document best practices.

  • Strong communication; can explain trade‑offs to non‑technical stakeholders.

  • Bias for shipping value in small, iterative increments.



Benefits

Our work timings are flexible, and we work five days a week.

Shift Boolean is an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to age, ancestry, color, family or medical care leave, gender identity or expression, genetic information, marital status, medical condition, national origin, physical or mental disability, political affiliation, protected veteran status, race, religion, sex (including pregnancy), sexual orientation, or any other characteristic protected by applicable laws, regulations, and ordinances. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. If you need assistance and/or a reasonable accommodation due to a disability during the application or the recruiting process, please send a request via the Accommodation request form.


Flexible Timing: 8 AM to 11 PM ( 8 Hours working ) Monday to Friday

Complete Remote Work from Home.




Salary:

INR 8 Lpa - 60 Lpa

Required profile

Experience

Level of experience: Junior (1-2 years)
Spoken language(s):
English
Check out the description to know which languages are mandatory.

Other Skills

  • Growth Mindedness
  • Communication

Artificial Intelligence Engineer Related jobs