Find jobs in AI/ML, Data Science and Big Data
12 results
for Flax
(Skill/Tech stack)
-
Sr. Software Engineer, AI Infrastructure USD 144K-236KApache Beam | Apache Flink | Apache Spark | C# | C++Career development | Hybrid work | Stock grantsSenior-level Full TimeSunnyvale, CA, United States3d ago
-
Deep learning | Experimental Design | Flax | Generative AI | JAXMid-level Full TimeMountain View, CA, USA; New York, …7d ago
-
Sr. Software Engineer, AI Infrastructure USD 139K-229KApache Beam | Apache Spark | C# | C++ | CUDAEmployee benefits | Hybrid work | Stock grantsSenior-level Full TimeSunnyvale, CA, United States7d ago
-
Research Scientist, Reinforcement Learning, DeepMind GBP 100K-180KAblation Studies | Algorithm Implementation | Deep learning | Evaluation | Experiment analysisMid-level Full TimeLondon, UK9d ago
-
Research Engineer, Human Understanding, DeepMind USD 174K-253KComputer Vision | Deep learning | Experimental Design | Flax | Generative AIMid-level Full TimeMountain View, CA, USA; Los Angeles, …14d ago
-
Research Scientist, Agent Post-Training, DeepMind GBP 100K-200KAgent-based | Agent-based systems | Experimentation | Fine Tuning | FlaxMid-level Full TimeLondon, UK16d ago
-
Data parallelism | Diffusion Models | Efficient Attention | Expert parallelism | FlaxSenior-level Full TimeMountain View, California, United States, New …24d ago
-
Computer Vision | Deep learning | Flax | Git | JAXBSPCE stock options | Remote work policySenior-level Full TimeParis, FR - Remote/Hybrid R25d 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, CA26d 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, Denmark29d 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, …1mo ago
-
C++ | Data Infrastructure | Embodied Agents | Flax | GPUMid-level Full TimeMountain View, CA, USA1mo ago