About this role
Research Software Engineer — Differentiable Scientific Computing (JAX/Julia) at Axiomatic_AI. Location: Boston or Cambridge or Barcelona. Role: building systems, optimizing pipelines, profiling workloads Requirements: 3+ years building scientific/numerical/HPC or ML infrastructure; deep expertise in JAX or Julia; strong Python/Julia engineering; numerical methods, AD, optimization, and profiling skills; clear communication. Category: Research and Development (R&D) Seniority: Mid Level Tools: JAX, Julia, Python, Kubernetes, Docker, PyVISA, SCPI, HPC, CI Commitment: Full Time Workplace: Hybrid Languages: English