"pytorch jax internals" Jobs - Joblaze

Showing 13 of 13 results

Member of Technical Staff (AI Inference Engineer)

Perplexity San Francisco Published 2 months ago

NVLink CuTe DSL Nsight Systems Triton JAX

Research Intern, Inference (Fall 2026)

Join Together AI as a Research Intern to work on cutting-edge distributed inference and optimization for large foundation models.

Together AI San Francisco $58–$63/yr Published 5 days ago

JAX CUDA Machine Learning PyTorch Python

Flexible on stack

Apply →

Frontier Agents Intern (Fall 2026)

Join Together AI as a research intern to work on cutting-edge projects in frontier AI systems and agentic behavior.

Together AI San Francisco $58–$63/yr Published 5 days ago

JAX Machine Learning PyTorch Python Transformer

Flexible on stack

Apply →

Research Intern, Model Shaping (Fall 2026) Remote

Join Together AI as a Research Intern to work on advanced machine learning techniques and contribute to impactful research.

Together AI San Francisco $58–$63/yr Published 2 days ago

JAX Machine Learning PyTorch Transformer Deep Learning

Flexible on stack

Apply →

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 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 →

Applied Research Intern Remote

Labelbox San Francisco Bay Area $35k–$45k/yr Published 10 months ago

Multimodal Models LLMs RLAIF JAX DPO

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 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 →

Research Engineer, Machine Learning (RL Velocity) Remote Visa

Anthropic Remote-Friendly (Travel-Required) | San Francisco, CA | New York City, NY $500k–$850k/yr Published 1 month ago

JAX AI/ML 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 →

Research Engineer, Discovery Remote Visa

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

Beam AWS Spark JAX AI/ML

Apply →