This Megatron-LM library (source https://github.com/NVIDIA/Megatron-LM) is adapted for use with AWS Trainium via Neuron SDK. Please see Neuron SDK documentation for tutorial on using Megatron-LM on AWS Trainium.
forked from aws-neuron/aws-neuron-reference-for-megatron-lm
-
Notifications
You must be signed in to change notification settings - Fork 0
License
GavinChen1314/aws-neuron-reference-for-megatron-lm
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
License
Code of conduct
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 89.2%
- C++ 7.5%
- Cuda 2.9%
- Other 0.4%