{"id":122473,"url":"https://alion.io/job/yottalabs-research-engineer-intern-ai-systems-2","title":"Research Engineer Intern - AI Systems","company":{"id":8214,"name":"Yotta Labs","domain":"yottalabs.com","url":"https://alion.io/company/yottalabs","size_band":null,"is_staffing_agency":false,"employer_type":"direct","is_intermediary":false,"listed_via":null,"ats_vendor":"Ashby","truth_index":{"grade":"B","score":75,"open_postings":3,"ghost_share":0,"stale_share":1,"repost_share":0,"time_to_fill_p50_days":null,"computed_at":"2026-10-21T05:46:15Z"}},"role":"AI/ML","role_family":"AI/ML","seniority":"intern","employment_type":"internship","work_mode":"remote","remote_scope":"stated_countries","remote_scope_basis":"board_field","remote_working_hours":null,"hiring_geo_confidence":"inferred","locations":["Hong Kong","Singapore"],"countries":["HK","SG"],"hiring_countries":["US","CA","HK","SG"],"hiring_countries_total":4,"salary":null,"salary_estimate":null,"experience_years_min":null,"visa_sponsorship":false,"relocation_package":false,"has_equity":false,"technologies":[{"name":"AWS","optional":false},{"name":"AWS Trainium","optional":false},{"name":"C++","optional":false},{"name":"CUDA","optional":false},{"name":"KV Cache","optional":false},{"name":"LLM","optional":false},{"name":"Mixture of Experts","optional":false},{"name":"Python","optional":false},{"name":"PyTorch","optional":false},{"name":"Quantization","optional":false},{"name":"ROCm","optional":false},{"name":"SGLang","optional":false},{"name":"Triton","optional":false},{"name":"vLLM","optional":false},{"name":"XLA","optional":false},{"name":"GitHub","optional":true},{"name":"Speculative Decoding","optional":true}],"status":"live","first_seen_at":"2026-08-02T16:29:42Z","employer_posted_date":"2026-08-02","last_verified_at":"2026-10-21T22:54:26Z","board_verified":true,"closed_at":null,"days_open":80,"trust":{"level":"ok","repost_count":null,"flags":[],"days_open":80},"description":"Location: Remote (Global)\nType: Internship\nCompany: Yotta Labs\nApply: \nAbout Yotta Labs\nYotta Labs is building the next generation multi-silicon AI cloud and runtime platform to power the world’s most demanding AI workloads. We enable training and inference across NVIDIA GPUs, AMD GPUs, and AWS Trainium, helping AI companies achieve the best performance and economics across heterogeneous hardware. Our mission is to provide high-performance AI computing and Model API services, enabling AI companies, research labs, and enterprises to train, deploy and integrate cutting-edge models at scale.\nRole Overview\nWe are seeking a highly motivated Research Engineer Intern to work on Trainium, GPU kernels, and LLM systems optimization. Over a 12-16 week internship, you will own a well-scoped project at the intersection of AI Systems, Compiler and Runtime Optimization, Distributed Training & Inference, GPU/Accelerator Kernel Development, and Large Language Model Infrastructure - taking it from design to working, profiled code running on real hardware. Your work will ship to production or open source and directly impact the performance of AI applications deployed on our platform. Strong interns receive return offers for full-time roles.\nResponsibilities\nImplement and optimize compute kernels for Attention, GEMM, MoE, and quantization on NVIDIA, AMD, or AWS Trainium.\n\nBuild custom operators using CUDA, Triton, ROCm/HIP, or the Neuron SDK with PyTorch/XLA.\n\nProfile and improve inference performance in vLLM, SGLang, and our custom runtimes - kernel fusion, scheduling, KV-cache and memory optimizations.\n\nBuild benchmarks, chase down performance regressions, and turn profiler traces into concrete speedups.\n\nShip code upstream to open-source AI infrastructure projects, with tests and documentation.\n\nQualifications\nCurrently pursuing a BS, MS, or PhD in Computer Science, Computer Engineering, or a related field.\n\nSolid programming skills in Python and familiarity with C++.\n\nUnderstanding of GPU/accelerator architecture fundamentals (memory hierarchy, parallelism, occupancy) from coursework, research, or projects.\n\nExperience writing CUDA, Triton, ROCm/HIP, or Neuron kernels - class projects and personal projects count.\n\nStrong understanding of AI frameworks (e.g., PyTorch, Dynamo, LMCache), model architectures and profiling tools (e.g. Nsight, ROCm Profiler, or Neuron Profiler).\n\nStrong problem-solving skills and the ability to work independently in a collaborative, remote environment.\n\nPreferred Experience\nContributions to open-source AI infra projects like vLLM, SGLang, PyTorch, or Triton.\n\nFamiliarity with LLM inference internals - FlashAttention, PagedAttention, continuous batching, speculative decoding, MoE, or quantization.\n\nExperience with profiling tools (e.g. Nsight, ROCm Profiler, Neuron Profiler, or PyTorch Profiler) and performance debugging on real workloads.\n\nPublications in top-tier conferences like MLSys, OSDI, SOSP, NSDI, SC, HPCA, or ISCA\n\nWhy Join Yotta Labs?\nBe part of a visionary team aiming to redefine AI infrastructure and influence the future of multi-silicon AI computing.\n\nWork on frontier AI infrastructure problems with access to serious hardware - latest-generation NVIDIA GPUs, AMD accelerators, and AWS Trainium at scale.\n\nGet direct mentorship from engineers from leading institutions and tech companies.\n\nCompetitive internship compensation, a flexible remote work environment, and a fast path to a full-time return offer for top performers.\n\nHow to Apply\nInterested candidates should apply directly or send their resume to . Please include links to any relevant projects or contributions (GitHub, open-source PRs, course projects) - for internships, these matter more to us than a cover letter.","description_format":"text","description_chars":3802,"description_truncated":false,"requirements":{"experience_years_min":null,"management_years_min":null,"team_size_min":null,"manages_managers":false,"education":{"level":"bachelor","optional":false},"security_clearance":false,"languages":[]},"benefits":[],"hiring_locations":[{"name":"United States","iso":"US","kind":"country"},{"name":"Canada","iso":"CA","kind":"country"},{"name":"Hong Kong","iso":"HK","kind":"country"},{"name":"Singapore","iso":"SG","kind":"country"}],"hiring_excludes":[],"relocation_offered":false,"industries":["Artificial Intelligence","Cloud Platforms (IaaS & PaaS)","AI Compute & Inference"],"lifecycle":[{"event":"open","at":"2026-08-02T16:29:42Z"}],"visa":[{"country":"US","licensed_sponsor":true,"evidence":"H-1B filings in 12 months: 1","filings_12m":1,"filings_prev_12m":1,"green_card_filings_12m":0,"median_offered_wage_usd":167500,"route":null,"cap_exempt":false,"checked_at":"2026-10-15T11:24:14+00:00","sources":["US Department of Labor: LCA disclosure data (H-1B, H-1B1, E-3)"],"filings_for_role_12m":1}],"liveness":{"score":5,"band":"cold","label":"Long shot","p_open":1,"p_active":0.191,"p_room":0.28,"age_days":79,"expected_fill_days":28,"reasons":["conf:1","win:tail","crowd:junior"],"computed_at":"2026-10-21T05:46:15Z"},"pay":null,"html_url":"https://alion.io/job/yottalabs-research-engineer-intern-ai-systems-2","json_url":"https://alion.io/job/yottalabs-research-engineer-intern-ai-systems-2.json","meta":{"generated_at":"2026-10-22T01:40:35Z","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":848,"day_limit":null,"remaining_today":null,"minute_limit":300,"resets_at":"2026-10-23T00: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":8214},"rest":"https://alion.io/mcp/rest/get_company?id=8214"},"human":"https://alion.io/catalog?offer=company.slices&for=job%2Fyottalabs-research-engineer-intern-ai-systems-2"},{"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%2Fyottalabs-research-engineer-intern-ai-systems-2"},{"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%2Fyottalabs-research-engineer-intern-ai-systems-2"},{"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/yottalabs-research-engineer-intern-ai-systems-2\", \"note\": \"what you need it for\"}"},"human":"https://alion.io/catalog?offer=company.verify&for=job%2Fyottalabs-research-engineer-intern-ai-systems-2"}]}