About this role
Member of Technical Staff - System Modeling (Computation) at Unconventional, Inc.. Location: Palo Alto or Remote. Role: Building large-scale GPU-accelerated solvers, Developing physics-based models with PyTorch, Collaborating with other teams Requirements: MS/PhD in AI/ML/CS/Physics/Electrical Engineering or BS with strong research depth; GPU kernel experience; DE solvers; neural surrogate models; PyTorch ML systems. Category: Research and Development (R&D) Seniority: Mid Level Tools: PyTorch, CUDA, Triton, JAX, NumPy, TensorFlow, Modal, MPI, NCCL, SciPy, TorchCompile Commitment: Full Time Workplace: Hybrid Languages: English