Machine Learning Engineer (EU Remote)

Job content

Note: This is a fully remote position, with preference for the EU timezones.About Kosen LabsWe are a team of ex-Google DeepMind AI researchers and engineers with extensive experience in designing and implementing cutting edge solutions to machine learning and numerical optimisation problems.The mission of Kosen Labs is to bring the benefits of AI & advanced optimization to the DeFi space and beyond, all while keeping entry barriers low for the end user and allowing everyone to partake in the new decentralised economy.We are funded by top-tier Silicon Valley and crypto-native VCs, such as Andreessen Horowitz and Framework Ventures.The RoleWe are looking for an ambitious ML Engineer who would feel immense bliss in their heart while reading sentences as follows:“With its updated version of Autograd, JAX can automatically differentiate native Python and NumPy functions.” “Others are vmap for automatic vectorization and pmap for single-program multiple-data (SPMD) parallel programming of multiple accelerators, with more to come.” “All it takes is 10-20 lines of code to get started with training a GNN model”. “TorchServe is an easy to use tool for deploying PyTorch models at scale. It is cloud and environment agnostic and supports features such as multi-model serving, logging, metrics and the creation of RESTful endpoints for application integration."If you felt the boogie while reading the above, apply now.RequirementsResponsibilitiesCode champion for the research team - balancing between being flexible with prototype code, but able to set standards for when it is being merged into our ‘serious’ repos/libraries that are shared between all team members.Act as a go between for researchers and data engineers, take our prototype and a) profile it b) error catch c) enforce coding standards d) help data team with productionisation. Working closely with researchers to enable research code development and refactoring, and writing more performant versions of our internal frameworks and libraries.Maintaining code quality and health and performing PR reviews.Helping with ML devops processes e.g. AMI setup, experiment management, making backtesting smoother, storing of backtest results/models in databases, etc.Requirements5+ years of software development experience in a solid company.Proven software architecture and design skills, especially when it comes to writing performant code.Adherence to coding standards and code quality, while at the same time being flexible with the needs of the research team.Solid experience of Python and Python data science stack (numpy, scipy, pandas).Good knowledge of at least one ML framework such as TensorFlow, PyTorch or JAX.Bonus pointsExperience with optimisation algorithms.Good C++ skills.Relevant ML research experience.Experience with distributed computing, accelerators (GPU, TPU), cloud computing environments.Experience with parallel computing and code optimization.BenefitsCompetitive salary;Equity and token allocation;Remote-first, flexible, and fun working environment;Unlimited time off;Occasional paid travel costs to conferences and team off-sites.