Find jobs in AI/ML, Data Science and Big Data
6 results
for Flax
(Skill/Tech stack)
-
Senior Research Scientist/Software Engineer, LLM/Agent Platform (TikTok-Content Ecology AI Innovation & Platform) USD 160K-230KCUDA | Deep learning | DeepSpeed ZeRO | Distributed Systems | Evaluation FrameworksSenior-level Full TimeSan Jose, California, United States3h ago
-
Staff ML Engineer, Perception Research USD 251K-310K3D Perception | Autoregressive models | Data Pipelines | Data parallelism | Diffusion Models401k match | Baby bonding leave | Dental insurance | Disability insurance | Health insuranceSenior-level Full TimeMountain View, CA, USA ; New … R13d ago
-
Data parallelism | Diffusion Models | Efficient Attention | Expert parallelism | FlaxSenior-level Full TimeMountain View, California, United States, New …1mo ago
-
Computer Vision | Deep learning | Flax | Git | JAXBSPCE stock options | Remote work policySenior-level Full TimeParis, FR - Remote/Hybrid R1mo ago
-
Research Scientist - Distributed Machine Learning USD 180K-287KBF16 | CUDA | CUDA kernels | DeepSpeed | Distributed Training401k | Dental insurance | Disability insurance | Employee assistance program | Health insuranceMid-level Full TimeSunnyvale, CA1mo ago
-
PhD scholarship in Causal-informed Machine Learning Methods for Climate Modelling - DTU Compute DKK 386K-386KCausal Inference | Diffusion Models | Flax | Graph Modeling | HPCFully funded PhD scholarship | International research environmentEntry-level Full TimeKgs. Lyngby, Denmark1mo ago