Find jobs in AI/ML, Data Science and Big Data
59 results
for JAX
(Skill/Tech stack)
-
Agile Development | Apache Beam | Data Analysis | Data Science | JAXBenefits | Bonus | EquityMid-level Full TimeMountain View, CA, USA; San Francisco, …1d ago
-
Software Engineer, Performance, YouTube ML Efficiency USD 141K-202KAlgorithms | Artificial Intelligence | Computer Architecture | Data Analysis | Data StorageBenefits | Bonus | EquityMid-level Full TimeSan Bruno, CA, USA1d ago
-
APIs | AWS | Azure | Cloud Native | Data PipelinesGym access | Healthcare | Lunch vouchers | Maternity leave | Paid time offSenior-level Full TimeParis, IDF, France2d ago
-
AI Research Engineer USD 300K-425KBlockchain | Cryptography | Decentralized networks | Distributed Systems | Federated LearningMid-level Full TimeAnywhere R2d ago
-
Quantum Software Engineer - Machine Learning USD 135K-160KBraket | Cirq | GPU | HPC | JAX401k | Dental | Disability insurance | Employee assistance | EquityMid-level Full TimeChicago, Illinois, United States3d ago
-
Code review | Deep learning | JAX | Motion Planning | Open SourceBenefits package | Competitive salariesSenior-level Full TimeChina, Shanghai3d ago
-
Senior AI/ML Engineer USD 158K-241KC++ | Computer Vision | Deep learning | JAX | ML deploymentDental | Health and wellbeing benefits | Holidays | Medical | Paid vacationSenior-level Full TimeMountain View Technical Center - Mountain …3d ago
-
Senior ML Compiler Engineer USD 152K-287KApache TVM | C++ | Deep learning | Deep learning frameworks | Flash AttentionBenefits | Competitive salaries | EquitySenior-level Full TimeUS, WA, Redmond, United States3d ago
-
Applied Machine Learning Engineer USD 110K-165KA3C | AWS | Azure | C++ | CUDA401k | Education assistance | Health care | Inclusive environment | Paid HolidaysSenior-level Full TimeColorado Springs, United States3d ago
-
Staff Machine Learning Engineer (Modeling), Support USD 276K-415KBatch inference | Classifiers | Experiment tracking | JAX | LLMsFamily planning | Flexible time off | Medical insurance | Remote work | Retirement plansSenior-level Full TimeSan Francisco, CA, United States of …3d ago
-
2026 Internship Project: Hamiltonian Learning USD 71K-130KData Analysis | JAX | Many-body physics | NumPy | Numerical simulationsCutting-edge research | Industry experience | Mentorship | Remote flexibilityEntry-level InternshipBerkeley, CA or Fremont, CA3d ago
-
AI/ML | Cloud deployment | Context engineering | Docker | FastAPIEntry-level Full TimeSeoul, Seoul, Korea, Republic of3d ago
-
Staff Software Engineer, ML Infrastructure USD 300K-430KBatching Strategies | Distributed Training | Fault Tolerance | Inference architecture | JAXDental | Lunches | Medical | Snacks | VacationSenior-level Full TimeSan Francisco4d ago
-
Research Scientist, AI-powered Scientific Discovery CAD 104K-193KCode synthesis | Deep learning | JAX | Language Models | Language ProcessingMid-level Full TimeMontreal, Canada4d ago
-
AI/ML | Cache Management | Cloud deployment | Context engineering | DockerEntry-level Full TimeGurgaon, Haryana, India4d ago
-
Senior AI Engineer, Computer Vision and Applied ML TWD 700K-1400KAR overlays | App Development | Application development | C++ | Camera managementSenior-level Full TimeNew Taipei, Banqiao District, New Taipei …4d ago
-
Senior Software Engineer, Camera Imaging TWD 2600K-3150KAlgorithm Development | C++ | Computer Vision | Digital Imaging | Embedded SystemSenior-level Full TimeNew Taipei, Banqiao District, New Taipei …4d ago
-
AI/ML Research Engineer, LLM Post-Training & Evaluation USD 170K-244KData Pipelines | Debugging | Distributed Training | Experiment tracking | Fine TuningAdvanced research projects | Collaborative environment | Opportunity for innovationSenior-level Full TimeRemote job R5d ago
-
Lead Software Engineer - AI I CAD 140K-175KAI Agents | AI Governance | AI Product Development | AI research | AI/MLCareer development | Employee incentive programs | Hybrid work model | Inclusive culture | Industry BenefitsSenior-level Full TimeCanada, Toronto, Ontario5d ago
-
Senior Data & AI Scientist GBP 70K-80KAPIs | Big Data | Cloud Platforms | Data Engineering | Deep learning30 days holiday | Annual bonus | Discounted shopping | Parental leave | Pension up to 15%Senior-level Full TimeLeeds Wellington Place, United Kingdom5d ago
-
AI | Agentic Architectures | Cloud Computing | Customer Engagement | DebuggingCareer growth | Flexible organization | Inclusive culture | Innovation environment | MentorshipSenior-level Full TimeNew York, New York, USA5d ago
-
Mid-level Full TimeIrving, TX, US5d ago
-
Ansible | C++ | CUDA | Cloud technologies | Distributed Training401k | Access to supercomputers | Disability insurance | Energy-efficient data centers | Flexible remote workSenior-level Full TimeSan Francisco5d ago
-
Member of Engineering (Pre-training / CUDA) USD 150K-250KAlgorithmic Skills | C/C++ | CUBLAS | CUDA | Critical ThinkingCompany equipment | Flexible hours | Health insurance | Remote work | Team get-togethersMid-level Full TimeRemote (EMEA/East Coast) R5d ago
-
AI Tooling | C/C++ | CUDA | Distributed Systems | Experimentation workflowsCareer growth | Cutting-edge tools | Equity | Health | Remote workMid-level Full TimeRemote; East Coast, US; Europe R5d ago
-
Machine Learning Engineer - IV (Biometrics) INR 1340K-1800KAWS | Airflow | Biometrics | Computer Vision | EC2Mid-level Full TimeBangalore5d ago
-
Senior software engineer, Machine Learning, Core ML USD 166K-244KC++ | Data Processing | Debugging | Deep learning | Deep learning frameworksBenefits | Bonus | EquitySenior-level Full TimeMountain View, CA, USA5d ago
-
Practice Customer Engineer, Cloud AI, Google Cloud USD 102K-146KAI Agent | AI accelerators | AI agent orchestration | API prompting | Agent OrchestrationBenefits | Bonus | EquityMid-level Full TimeSunnyvale, CA, USA; San Francisco, CA, …5d ago
-
Practice Customer Engineer, Cloud AI, Google Cloud USD 102K-146KAI Agent | AI accelerators | AI agent orchestration | AI agent orchestration frameworks | API promptingBenefits | Bonus | EquityMid-level Full TimeSunnyvale, CA, USA; San Francisco, CA, …5d ago
-
Practice Customer Engineer, Cloud AI, Google Cloud USD 102K-146KAI Agent | AI accelerators | AI accelerators TPU | AI agent orchestration | API promptingBenefits | Bonus | EquityMid-level Full TimeSunnyvale, CA, USA; Seattle, WA, USA5d ago
-
Senior Data Scientist / ML Engineer - OPS00052 BRL 234K-295KDebugging | Deep learning | Gradient optimization | JAX | Machine LearningEducational support | Paid rest days | Remote HR and payroll support | Team-building activitiesSenior-level Full TimeSão Paulo, São Paulo, Brazil - … R6d ago
-
Machine Learning Engineer 50 USD 172K-306KAI workflows | AWS | Azure | Cloud Platforms | Data AnalysisSenior-level Full TimeSan Jose, United States6d ago
-
Research Scientist, AQUA INR 2500K-4500KCollaboration | Communication | Deep learning | Deployment | JAXSenior-level Full TimeBangalore, India7d ago
-
Principal Engineer, Databases Agentic AI USD 294K-414KAI Innovation | AI Model Evaluation | AI capabilities | AI capabilities in databases | AI infrastructureBenefits | Bonus | EquitySenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA7d ago
-
Summer Internship - Machine Learning Engineer USD 72K-96KAlgorithms | Analytical Skills | C++ | Communication skills | Data StructuresCollaborative team environment | Hands-on project experienceEntry-level InternshipAustin, TX8d ago
-
Staff Software Engineer, Generative AI, Core ML USD 197K-291KBenchmarking | Data Processing | Debugging | Embodied Agents | Evaluation harnessesBenefits | Bonus | EquitySenior-level Full TimeMountain View, CA, USA9d ago
-
Software Engineer III, Google Research, AI/ML USD 141K-202KArtificial Intelligence | C++ | Deep learning | Deep learning frameworks | Information RetrievalBenefits | Bonus | EquitySenior-level Full TimeMountain View, CA, USA9d ago
-
AI frameworks | Data analytics | Debugging | Distributed Training | JAXBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara, United States10d ago
-
Senior Deep Learning Algorithm Engineer USD 184K-356KDebugging | Distributed Computing | JAX | Mixed Precision | Mixed-precision trainingBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara, United States10d ago
-
AI Engineer - NY USD 40K-141KAWS | Agent systems | Airflow | Azure | CI/CD401k | Dental | Medical | Paid Holidays | Paid time offMid-level Full TimeUnited States10d ago
-
Staff Machine Learning Engineer USD 210K-275KCloud Environments | Data Pipelines | Deep learning | JAX | Machine LearningSenior-level Full TimeMountain View, CA11d ago
-
Senior-level Full TimeMountain View, CA11d ago
-
Machine Learning Engineer USD 155K-207KAI research | Cloud infrastructure | Data Pipelines | Deep learning | JAXMid-level Full TimeMountain View, CA11d ago
-
Mid-level Full TimeNew York, New York, United States11d ago
-
AI frameworks | AI frameworks (PyTorch | CI/CD | CUDA Python | Collaborative engineeringBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara, United States11d ago
-
Software Development Engineer USD 159K-301KAI systems | Architecture | CUDA | Data Engineering | Deep learningSenior-level Full TimeSan Jose, United States12d ago
-
Applied Atomistic Modeling Researcher, Drug Discovery USD 192K-304KAtomistic modeling | C/C++ | CUDA | Computational Biology | Conference presentationBenefits | EquityMid-level Full TimeUS, CA, Santa Clara, United States12d ago
-
C++ | CUDA | Data Structures | Distributed Learning | GPU ProgrammingBenefits | Competitive salary | EquitySenior-level Full TimeUS, CA, Santa Clara, United States12d ago
-
AI for Science | Atomistic modeling | CUDA | Deep understanding | Deep understanding of compute bottlenecksAI-focused environment | Competitive salaries | Comprehensive benefits | EquitySenior-level Full TimeUS, CA, Santa Clara, United States12d ago
-
Data Engineer - Training Pipelines & Inference USD 86K-183KAWS SageMaker | Big Data | Big Data Technologies | Cloud Computing | Data ProcessingCollaborative environment | Cutting-edge scientific projects | Health benefits | On-site amenities | Professional developmentMid-level Full TimeJanelia Research Campus, United States12d ago