Tengxu-Sun

bagua

Bagua Speeds up PyTorch

long-context-attention

USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference

CoreML-Custom-Layers

Source code for the blog post "Custom Layers in Core ML"

bagua

Bagua is a flexible and performant distributed training algorithm development framework.

CoreML-Custom-Layers

Source code for the blog post "Custom Layers in Core ML"

Pytorch-Topk-Bug

pytroch-coreml-custom-layer-example

PyTorch to CoreML: Writing custom layers with Metal

smilelogging

Python logging package for easy reproducible experimenting in research @ SMILE Lab

torch-mlir

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.

tutorials

Bagua tutorials.

Tengxu-Sun/bagua

Bagua is a flexible and performant distributed training algorithm development framework.

Tengxu-Sun/CoreML-Custom-Layers

Source code for the blog post "Custom Layers in Core ML"

Tengxu-Sun/Pytorch-Topk-Bug

Tengxu-Sun/pytroch-coreml-custom-layer-example

PyTorch to CoreML: Writing custom layers with Metal

Tengxu-Sun/smilelogging

Python logging package for easy reproducible experimenting in research @ SMILE Lab

Tengxu-Sun/torch-mlir

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.

Tengxu-Sun/tutorials

Bagua tutorials.

Tengxu-Sun/book

个人认为对技术提升很不错的书