"pytorch jax internals" Jobs - Joblaze

Showing 14 of 14 results

Research Engineer, Interpretability Remote Visa

Anthropic San Francisco, CA $315k–$560k/yr Published 7 months ago

JAX CUDA Interpretability Java Go

Apply →

Research Engineer, Pretraining Scaling - London Visa

Anthropic London, UK £260k–£630k/yr Published 8 months ago

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

Apply →

Research Engineer, Pretraining Scaling Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 8 months ago

ML frameworks JAX model operations AI/ML large-scale distributed systems

Apply →

Research Engineer/Research Scientist, Audio Remote Visa

Anthropic San Francisco, CA $350k–$500k/yr Published 5 months ago

JAX Large Language Models Diffusion Models Audio Codecs reinforcement learning

Apply →

Research Scientist, AQUA Visa

Google DeepMind Bangalore, India Published 3 months ago

JAX Machine Learning AI/ML reinforcement learning PyTorch

Apply →

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

Anthropic San Francisco, CA | New York City, NY $500k–$850k/yr Published 1 year ago

C++ Trio JAX async/concurrent programming Rust

Apply →

Research Scientist, Reinforcement Learning

Google DeepMind London, UK Published 3 months ago

JAX AI/ML reinforcement learning PyTorch Flax

Apply →

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

Anthropic London, UK £260k–£630k/yr Published 4 months ago

C++ Trio JAX async/concurrent programming Rust

Apply →

Research Scientist – Controlled 3D Generation Remote

Stability AI Remote Published 7 months ago

Flow Matching JAX Geometry Processing CUDA Machine Learning

Apply →

Research Engineer, Performance RL (Reinforcement Learning) Remote Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 2 months ago

Triton JAX CUDA reinforcement learning PyTorch

Apply →

ML Infrastructure Engineer, Safeguards Remote Visa

Anthropic San Francisco, CA $320k–$405k/yr Published 11 months ago

AWS Spark JAX Large Language Models transformer architectures

Apply →

Sr. Software Engineer (Vehicle Engineering) Visa

Join SpaceX as a Senior Software Engineer to develop AI systems that enhance engineering processes for launch vehicles and spacecraft.

SpaceX Hawthorne, CA $160k–$225k/yr Published 1 month ago

JAX Machine Learning reinforcement learning PyTorch TensorFlow

70% coding

Apply →

Senior Site Reliability Engineer, AI Research Remote

Algolia Remote - Australia Published 4 months ago

Terraform Google Cloud Platform Datadog CI/CD ArgoCD

Apply →

Research Engineer, Discovery Remote Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 1 year ago

Beam AWS Spark JAX AI/ML

Apply →