SherlockNoMad/ate-bench
Evaluate PithTrain, Megatron-LM, TorchTitan on agent-task efficiency.
Evaluate PithTrain, Megatron-LM, TorchTitan on agent-task efficiency.
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Sample ccript for running PrimsTorch+ nvFuser stack
Deep Learning Examples
📖The Big-&-Extending-Repository-of-Transformers: Pretrained PyTorch models for Google's BERT, OpenAI GPT & GPT-2, Google/CMU Transformer-XL.
Open Neural Network Exchange
Create2 library for python