← 返回名录条目信息来自其公开主页与公开发布内容。按本站规范，页面不展示任何联系方式。 需要更正或删除？通过收录与更正通道提交，24 小时内处理。
S
Shaogang Wang
Machine learning compiler engineer at NVidia.
- 公司
- nvidia
- 位置
- Changsha, China
- Stars
- 0
- 粉丝 / 仓库
- 6
代表作品 / 项目
- JAX-Toolbox⭐ 0JAX-Toolbox
- nccl⭐ 0Optimized primitives for collective multi-GPU communication
- TransformerEngine⭐ 0A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
- nccl_test⭐ 0
- paxml⭐ 0Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.