{"id":3466162,"url":"https://alion.io/job/cubicmind-ai-junior-ml-engineer-model-architecture-training-and-edge-optimisation","title":"Junior+ ML Engineer (Model Architecture, Training and Edge Optimisation)","company":{"id":4486757,"name":"Cubicmind Ai","domain":"cubicmind-ai.com","url":"https://alion.io/company/cubicmind-ai","size_band":null,"is_staffing_agency":false,"employer_type":"direct","is_intermediary":false,"listed_via":null,"ats_vendor":null,"truth_index":null},"role":"AI/ML","role_family":"AI/ML","seniority":"junior","employment_type":"full_time","work_mode":"remote","remote_scope":"stated_countries","remote_scope_basis":"board_field","remote_working_hours":null,"hiring_geo_confidence":"inferred","locations":["Ukraine"],"countries":["UA"],"hiring_countries":["UA"],"hiring_countries_total":1,"salary":{"min":1000,"max":1500,"currency":"USD","period":"month","gross":false,"usd_annual":18000},"salary_estimate":null,"experience_years_min":null,"visa_sponsorship":false,"relocation_package":false,"has_equity":false,"technologies":[{"name":"C++","optional":false},{"name":"Knowledge Distillation","optional":false},{"name":"KV Cache","optional":false},{"name":"Machine Learning","optional":false},{"name":"Model Distillation","optional":false},{"name":"Post-training","optional":false},{"name":"PyTorch","optional":false},{"name":"Quantization","optional":false},{"name":"Transformers","optional":false},{"name":"ONNX","optional":true},{"name":"OpenVINO","optional":true},{"name":"TensorRT","optional":true}],"status":"live","first_seen_at":"2026-10-16T02:26:09Z","employer_posted_date":null,"last_verified_at":"2026-10-16T02:26:09Z","board_verified":false,"closed_at":null,"days_open":14,"trust":{"level":"not_scored","repost_count":null,"flags":[],"days_open":14},"description":"Full-time\n\nSalary: $1000—$1,500/month, reviewed after 3 months based on delivery\nPosition is for recent university graduates majored in Machine Learning, Mathematics or related field.\nPrior work experience is not required, but deep understanding of machine learning and mathematics under the bonnet is.\n\nAbout CubicMind AI\n\nCubicMind AI Ltd is a UK deep tech startup building a small plug-in device that operates any computer like a person: it reads the screen, listens to the user, and clicks and types to complete tasks. Everything runs on the device itself, with no cloud.\ncubicmind-ai.com\n\nWe design our own models rather than wrapping someone else’s. That includes transformer and non-transformer architectures, built around what runs fast on edge hardware, and sometimes trained from scratch.\n\nAbout the role\n\nYou will work on designing, training and optimising the models that power the device: screen understanding, action prediction and efficient on-chip inference.\n\nThis is research-heavy engineering. You will work on model architecture, training runs, attention and memory bottlenecks, low-bit compression, and getting models to run correctly on real hardware.\n\nWhat you will do\n\n— Design and implement new model architectures in PyTorch, both transformer-based and beyond\n\n— Train models, sometimes from scratch, and run experiments to compare architectures\n\n— Build training and evaluation pipelines on screen-understanding and computer-control data\n\n— Profile latency, memory use and KV-cache behaviour on long, high-resolution screen inputs\n\n— Compress models for edge chips using quantisation (INT8/INT4), pruning and distillation\n\n— Export models to low-level runtimes and fix unsupported operators\n\n— Work with our C++ team to get models running efficiently on the device\nRequirements\n\n— Architectural understanding of Transformers and Attention: for example you understand the mathematical and operational differences between MHA, MQA, GQA, FlashAttention, and linear attention variants; \n\n— Low-level PyTorch fluency: writing custom PyTorch modules, vectorised tensor operations, e.g. knowing how to do ML\n\n— Experience with model training & convergence: stability tricks, learning rate schedules, gradient clipping, mixed-precision training (BF16/FP16), and diagnosing loss spikes or exploding/vanishing gradients.\n\n— Mathematical foundation: comfortable with high-dimensional linear algebra, matrix calculus, multivariable optimization, and probability theory.\n\n— Understanding what model compression is: Post-Training Quantization vs Quantization-Aware Training, outlier activations, etc\n\n— Сommit full-time\nNice to have\n\n— Experience with vision transformers or vision-language models\n\n— Interest in non-transformer architectures (state-space models, linear attention, recurrent designs)\n\n— Experience exporting to ONNX, TensorRT, OpenVINO or NPU toolchains\n\n— Basic C++, enough to read inference code\nWhat we offer\n\n— Real ownership of the core models in a hardware product built from the ground up\n\n— Work directly with the founder, with no bureaucracy\n\n— $1000—$1,500/month depending on experience, reviewed after 3 months based on results\n\n— Fully remote, judged on results, not hours logged. Flexible schedule","description_format":"text","description_chars":3239,"description_truncated":false,"requirements":{"experience_years_min":null,"management_years_min":null,"team_size_min":null,"manages_managers":false,"education":null,"security_clearance":false,"languages":[]},"benefits":["Flexible schedule"],"hiring_locations":[{"name":"Ukraine","iso":"UA","kind":"country"}],"hiring_excludes":[],"relocation_offered":false,"industries":[],"lifecycle":[{"event":"open","at":"2026-10-29T15:24:45Z"}],"visa":[],"liveness":{"score":62,"band":"ok","label":"Likely open","p_open":0.85,"p_active":0.815,"p_room":0.9,"age_days":14,"expected_fill_days":25,"reasons":["seen:14","win:mid","comp:junior"],"computed_at":"2026-10-30T05:48:23Z"},"pay":{"stated_usd_annual":18000,"is_top_pay":false},"html_url":"https://alion.io/job/cubicmind-ai-junior-ml-engineer-model-architecture-training-and-edge-optimisation","json_url":"https://alion.io/job/cubicmind-ai-junior-ml-engineer-model-architecture-training-and-edge-optimisation.json","meta":{"generated_at":"2026-10-30T22:58:23Z","cache_seconds":300,"methodology":"https://alion.io/methodology","terms":"https://alion.io/terms","contact":"https://alion.io/contact","api":"https://alion.io/developers","about":"Alion is a live layer of people, companies and AI agents: who they are, whether they are real and active right now, what they do and how to work with them, readable by people and by agents and paid per call.","catalog":"https://alion.io/catalog.json","usage":{"tier":"crawler_verified","counted_by":"address","units_charged":1,"used_today":14025,"day_limit":null,"remaining_today":null,"minute_limit":300,"resets_at":"2026-10-31T00:00:00Z"}},"offers":[{"id":"company.slices","title":"One company in depth, by slice","status":"live","price":{"credits":0.02,"usd":0.002,"plus_per_slice":{"credits":0.05,"usd":0.005}},"unit":"per company, plus each slice with data","note":"the employer in depth","call":{"mcp_tool":"get_company","arguments":{"id":4486757},"rest":"https://alion.io/mcp/rest/get_company?id=4486757"},"human":"https://alion.io/catalog?offer=company.slices&for=job%2Fcubicmind-ai-junior-ml-engineer-model-architecture-training-and-edge-optimisation"},{"id":"market.stats","title":"A market slice: pay, demand and time to fill","status":"live","price":{"credits":1,"usd":0.1},"unit":"per slice","note":"pay, demand and time to fill for this role and place","call":{"mcp_tool":"market_stats"},"human":"https://alion.io/catalog?offer=market.stats&for=job%2Fcubicmind-ai-junior-ml-engineer-model-architecture-training-and-edge-optimisation"},{"id":"job.search","title":"Open jobs by role, technology, place, pay and visa","status":"live","price":{"credits":0.02,"usd":0.002},"unit":"per posting in a list","note":"similar open postings","call":{"mcp_tool":"search_jobs"},"human":"https://alion.io/catalog?offer=job.search&for=job%2Fcubicmind-ai-junior-ml-engineer-model-architecture-training-and-edge-optimisation"},{"id":"company.verify","title":"Is this company real and active right now","status":"pilot","price":null,"unit":"per company","request":{"url":"https://alion.io/catalog/request","method":"POST","body":"{\"offer\": \"company.verify\", \"for\": \"job/cubicmind-ai-junior-ml-engineer-model-architecture-training-and-edge-optimisation\", \"note\": \"what you need it for\"}"},"human":"https://alion.io/catalog?offer=company.verify&for=job%2Fcubicmind-ai-junior-ml-engineer-model-architecture-training-and-edge-optimisation"}],"agent":{"mcp":"https://alion.io/mcp/jobs","mcp_open":"https://alion.io/mcp/open","next":[{"tool":"get_job","arguments":{"id":3466162}},{"tool":"search_jobs","arguments":{"company":"cubicmind-ai.com","detail":"compact"}},{"tool":"match_people","arguments":{"job_id":3466162},"note":"people open to offers who fit this vacancy"}],"note":"The same data through MCP tools: https://alion.io/mcp/open needs no sign-in (compact searches, postings, people); https://alion.io/mcp with an Alion account has everything. Docs: https://alion.io/developers"}}