Állás

Kutatómérnök

AI Research Scientist • Helyszíni • Teljes munkaidő • Amerikai Egyesült Államok New York, USA

A Lightning AI kutatómérnökként a tréning- és inferencia-munkaterhelések optimalizálásáért felelsz a Lightning AI infrastruktúráján, az ML-rendszerek, AI-infrastruktúra és teljesítménymérnökség metszéspontjában.

Stack

Feladatok

  • ▹Nagy léptékű tréning- és inferencia-munkaterhelések optimalizálása GPU-kon, gyorsítókon és elosztott rendszereken
  • ▹Közvetlen együttműködés ügyfelekkel: munkaterhelések elemzése, szűk keresztmetszetek azonosítása, teljesítmény és megbízhatóság javítása
  • ▹Inferencia-pipeline-ok, modellkiszolgáló rendszerek és teljesítmény-orientált eszközök fejlesztése éles AI-munkaterhelésekhez
  • ▹Profilozó, hibakereső és observability-eszközök tervezése és megvalósítása modellvégrehajtás elemzésére
  • ▹Munka a teljes szoftverstacken, hogy a teljesítményjavítások tiszta API-kon és automatizáláson keresztül elérhetők legyenek
  • ▹Együttműködés hardvergyártókkal és ökoszisztéma-partnerekkel a hatékony végrehajtás támogatásában (NVIDIA, TPU, új gyorsítók)
  • ▹Hozzájárulás nyílt forráskódú projektekhez: új funkciók, eszközfejlesztés, dokumentáció, közösségi részvétel

Elvárások

  • ▹Erős szakértelem mélytanulási keretrendszerekben, elsősorban PyTorch
  • ▹Tapasztalat nagy léptékű tréning vagy inferencia munkaterhelésekkel
  • ▹Ismeretség elosztott rendszerekkel és párhuzamosítási stratégiákkal (data/model/pipeline parallelism, checkpointing, elasztikus skálázás, elosztott inferencia)
  • ▹Erős szoftvermérnöki alapok: API-tervezés, eszközépítés, komplex rendszerek hibakeresése, éles minőségű kód szállítása
  • ▹Tapasztalat teljesítmény-szűk-keresztmetszetek elemzésében és javításában ML-rendszerekben vagy elosztott munkaterhelésekben
  • ▹Kiváló együttműködési és kommunikációs készségek, kereszt-funkcionális és külső partnerekkel is
  • ▹Képesség bizonytalan, gyorsan változó környezetben, a stack több rétegén átívelő munkavégzésre
  • ▹Alapképzés (BSc) informatika, mérnöki vagy kapcsolódó területen

Előny

  • ▹Tapasztalat inferencia-optimalizálási technikákkal: kvantálás, speculative decoding, mixed precision, memóriahatékony tréning, throughput/latency optimalizálás
  • ▹Tapasztalat technológiákkal: CUDA, Triton, TensorRT, vLLM, SGLang, Dynamo vagy hasonló ML-rendszer/inferencia-eszközök
  • ▹Hozzájárulás nyílt forráskódú ML-, infrastruktúra- vagy AI-rendszer projektekhez
  • ▹Startup-tapasztalat vagy erősen kereszt-funkcionális környezetben szerzett tapasztalat
  • ▹MSc vagy PhD fokozat AI, gépi tanulás, rendszerek vagy kapcsolódó területen

Soft skillek

Kiváló együttműködés és kommunikáció, ügyfelekkel és külső partnerekkel isHelytállás bizonytalan, gyorsan változó környezetbenGyakorlatias, rendszerszemléletű problémamegoldásNyitottság a nyílt forráskódú közösségi munkára

Amit kínálunk

  • ▹Becsült éves alapbér 120 000-250 000 USD tartományban, diszkrecionális bónusszal és jelentős részvényjuttatással
  • ▹Teljes körű orvosi, fogászati és látásbiztosítás (USA); magán orvosi és fogászati biztosítás (UK)
  • ▹Nyugdíj- és pénzügyi jólléti támogatás (USA); nyugdíjhozzájárulás (UK)
  • ▹Bőkezű fizetett szabadság és ünnepnapok
  • ▹Fizetett szülői szabadság
  • ▹Szakmai fejlődési támogatás
  • ▹Wellness és otthoni munkavégzési keret
  • ▹Rugalmas munkavégzési környezet

A munkáltatóról

A Lightning AI a PyTorch Lightning mögött álló vállalat, amely 2019 óta épít end-to-end platformot AI-rendszerek fejlesztésére, tréningezésére és üzembe helyezésére. A Voltage Parkkal történt egyesülés után fejlesztői-központú szoftvert kombinál költséghatékony, nagy léptékű számítási kapacitással. Irodái New Yorkban, San Franciscóban, Seattle-ben és Londonban vannak, olyan befektetők támogatják, mint a Coatue, az Index Ventures, a Bain Capital Ventures és a Firstminute.

Nyelvtudás: Angol: Felsőfok
Végzettség: Alapképzés (BSc) informatika, mérnöki vagy kapcsolódó területen

Hasonló állások