Senior Software Engineer - Python Numerical Computing Libraries
Tasks
- Analyze and improve API performance on CPU and GPU
- Architect and develop accelerated distributed numerical algorithms
- Contribute to design and development of accelerated distributed Python APIs
- Design Python APIs for accelerated numerical scientific computing
- Develop runtime systems for multi GPU computing
- Prototype integrations into targeted frameworks
- Write maintainable production code
Perks/Benefits
Skills/Tech-stack
Array computing | Asynchronous programming | C++ | CUDA | Compiler optimization | CuNumeric | Cupy | Dense Array Computing | Distributed Computing | GPU Performance | GPU Performance Optimization | JAX | Multithreading | NumPy | Numba | Pandas | Parallel Computing | Performance Analysis | Performance optimization | Profiling | PyTorch | Python | SciPy | Scikit-learn | Sparse Array Computing | TensorFlow
Education
Regions
Countries
States
Related jobs
-
Software Engineer (Applied AI/ML) USD 200K-250KCI/CD | Data Pipelines | Data ontologies | Design of Experiments | Digital Twins401k | Bonuses | Equity | FSA | Flexible time offMid-level Full TimeSanta Clara, CA or Remote R3d ago
-
Senior Software Engineer, RL Post-Training Frameworks USD 184K-287KActor Based Programming | C# | C++ | Deep learning | Distributed SystemsSenior-level Full TimeUS, CA, Santa Clara R3d ago
-
Senior-level Full TimeUS, CA, Santa Clara R4d ago
-
Senior AI Infrastructure Software Engineer - DGX Cloud USD 184K-287KC# | C++ | Cause analysis | Debugging | Distributed SystemsBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R4d ago
-
Senior-level Full TimeUS, CA, Santa Clara R5d ago
-
BEV | C++ | Computer Vision | Data Augmentation | Data SimulationBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R6d ago
-
Senior Deep Learning Algorithm Engineer USD 184K-287KCPU architecture | Distributed Computing | GPU Architecture | Generative AI | JAXEquity | Health benefitsSenior-level Full TimeUS, CA, Santa Clara R10d ago
-
Senior Deep Learning Compiler Engineer - XLA USD 152K-241KC# | C++ | CUDA | Code generation | DebuggingSenior-level Full TimeUS, CA, Santa Clara R10d ago
-
C++ | CUDA | CuDF | DataFrame | Database accelerationSenior-level Full TimeUS, CA, Santa Clara R10d ago
-
C++ | CUDA | Docker | Infiniband | JAXSenior-level Full TimeUS, CA, Santa Clara R10d ago
-
Senior Software Engineer, AI Performance Analysis USD 184K-287KAI | C++ | Integration | Machine Learning | Performance AnalysisEmployee benefits | EquitySenior-level Full TimeUS, CA, Santa Clara R12d ago
-
Distinguished Engineer, Scaled Out Inferencing USD 320K-488KAI infrastructure | CUDA | Cloud Native | Cloud Native Architecture | Container OrchestrationSenior-level Full TimeUS, CA, Santa Clara R13d ago
-
Software Engineer, CUDA Deep Learning Systems USD 152K-258KC++ | CUDA | Computer Architecture | Deep learning | Distributed ComputingBenefits | EquityMid-level Full TimeUS, CA, Santa Clara R13d ago
-
C++ | CUDA | CUPTI | Continuous batching | KV cacheSenior-level Full TimeUS, CA, Santa Clara R18d ago
-
AWS | Azure | Bazel | C# | C++Equity | Hybrid workSenior-level Full TimeUS, CA, Santa Clara R18d ago
-
C# | C++ | Caching | Distributed Systems | GPU ArchitectureComprehensive benefits | EquitySenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Senior Software Engineer – TensorRT Edge-LLM USD 152K-241KAttention | C++ | CUDA | Compiler optimization | FlashInferSenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Senior Software Engineer, CUDA Deep Learning Systems USD 184K-356KC++ | CUDA | Computer Architecture | Deep Learning Compilers | Deep learningBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Senior Deep Learning Algorithm Engineer USD 152K-241KAutoscaling | C++ | Deep learning | Distributed Systems | Inference OptimizationSenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Senior Storage Production Engineer - DGX Cloud USD 184K-333KAccess Control | Anomaly Detection | Ansible | Auditing | BashSenior-level Full TimeUS, CA, Santa Clara R19d ago
-
C++ | CPU architecture | CUDA | Deep learning | Fraud DetectionSenior-level Full TimeUS, CA, Santa Clara R19d ago
-
Bash | Bootstrap | CSI | CSS3 | Container StorageSenior-level Full TimeUS, CA, Santa Clara R20d ago
-
Senior Software Engineer, CUDA Deep Learning Systems USD 184K-356KC++ | CUDA | CUDA kernels | Computer Architecture | Deep learningBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R21d ago
-
Senior Deep Learning Frameworks CUDA Software Engineer USD 184K-356KC++ | CUDA | Distributed machine learning | HPC | JAXSenior-level Full TimeUS, CA, Santa Clara R21d ago
-
Senior Deep Learning Framework Communications Engineer USD 184K-287KC++ | CUDA | CuTe | Elasticity | Fault ToleranceBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R21d ago