About the Role

We are hiring a Senior Python Engineer with strong scientific computing depth for a backend role focused on algorithmic libraries. You will work at the intersection of applied mathematics, probabilistic modeling, and production-grade Python - helping us build and maintain robust, well-tested scientific backends.

Must Have

- Python 3.11+ : Type hints, pydantic, dataclasses - strong typing discipline required.

- JAX, NumPy, SciPy : Hands-on production experience with scientific computing libraries.

- Probabilistic ML : Working knowledge of POMDPs, HSMMs, or Bayesian state-space models.

- Testing & Quality : pytest, mypy strict mode, dependency hygiene with uv or Poetry.

- Algorithmic Code : Comfortable reading and documenting complex algorithmic code.

Good to Have :

- pymdp / interactively-pymdp : Experience with Active Inference tooling for Markov Decision Processes.

- pgmpy : Familiarity with probabilistic graphical models and Bayesian network libraries.

- Refactoring Experience : Experience refactoring server-side libraries to remove embedded UI dependencies.

What You'll Be Doing

- Maintain and extend algorithmic Python libraries used in production scientific pipelines.

- Write clean, well-typed, and thoroughly tested code following strict mypy and pytest standards.

- Collaborate with ML researchers to implement and optimize probabilistic models.

- Refactor legacy codebases to improve modularity, remove UI coupling, and improve testability.

- Document complex mathematical and algorithmic code for team-wide clarity.

- Immediate Joiners Preferred - Full-time - 1 Position Only

Only candidates who can join immediately will be considered for this role.