Find jobs in AI/ML, Data Science and Big Data
10 results
for Flax
(Skill/Tech stack)
-
Research Scientist, Reinforcement Learning, DeepMind GBP 100K-180KAblation Studies | Data Analysis | Deep learning | Evaluation | Experiment designMid-level Full TimeLondon, UK9h ago
-
Research Scientist, Agent Post Training, DeepMind USD 174K-253KData Analysis | Distributed Training | Experimentation | Flax | Human FeedbackMid-level Full TimeMountain View, CA, USA9h ago
-
Research Scientist, Agent Post-Training, DeepMind GBP 100K-200KDistributed Training | Flax | Human Feedback | JAX | Large Scale TrainingMid-level Full TimeLondon, UK9h ago
-
Data parallelism | Diffusion Models | Efficient Attention | Expert parallelism | FlaxSenior-level Full TimeMountain View, California, United States, New …3d ago
-
Computer Vision | Deep learning | Flax | Git | JAXBSPCE stock options | Remote work policySenior-level Full TimeParis, FR - Remote/Hybrid R4d 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, CA5d 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, Denmark8d ago
-
Senior Machine Learning Engineer, Simulation Evaluation USD 213K-263K4D modeling | C++ | Computer Vision | Data Pipelines | Deep learningBonus program | Company benefits program | Equity incentive planSenior-level Full TimeMountain View, CA, USA; San Francisco, …18d ago
-
3D Computer Vision | 3D Gaussian Splatting | 3D Reconstruction | C++ | Computer VisionSenior-level Full TimeMountain View, CA, USA24d ago
-
C++ | Data Infrastructure | Embodied Agents | Flax | GPUMid-level Full TimeMountain View, CA, USA24d ago