About this role
Post-Training Research Engineer at Baseten. Location: San Francisco, California, United States. Role: build tooling, train transformers, profile performance Requirements: Strong ML fundamentals, transformer training, distributed GPU, HPC, and systems-level skills including Kubernetes and PyTorch or similar. Category: Engineering Seniority: Mid Level Tools: PyTorch, TensorFlow, Jax, Kubernetes, Slurm, Ray, Dask, GPUDirect, Infiniband Commitment: Full Time Workplace: Hybrid Languages: English