Skip to content

Commit d412558

Browse files
committed
docs: Update readme
1 parent 3b52070 commit d412558

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

README.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@ For more details, please refer to my blog post: [Write a nn training framework f
1212

1313
## Key Features
1414

15-
* **PyTorch-Style API**: Adopts similar class and function naming conventions as PyTorch, such as `Tensor`, `Functions`, `nn.Module`, and `Optimizer`, to make it intuitive for users familiar with the original framework.
16-
* **Pure C++ Implementation**: The entire framework is written in C++ with no dependency on external deep learning libraries, making it perfect for understanding the fundamentals.
17-
* **CPU and CUDA Support**: The framework is built to run on both **CPU** and **CUDA** enabled GPUs, allowing for flexible development and experimentation.
18-
* **Data Type Support**: The framework supports various data types, including float32, float16, and bfloat16.
19-
* **LLM Inference**: Supports inference for gpt2 and llama3.2 models: [https://github.com/keith2018/TinyGPT](https://github.com/keith2018/TinyGPT)
15+
* **PyTorch-Style API**: Adopts similar class and function naming conventions as PyTorch, such as `Tensor`, `Functions`, `nn.Module`, and `Optimizer`.
16+
* **Pure C++ Implementation**: The entire framework is written in C++ with no dependency on external deep learning libraries.
17+
* **CPU and CUDA Support**: The framework is built to run on both **CPU** and **CUDA** enabled GPUs.
18+
* **Mixed Precision Support**: The framework supports multiple precisions such as FP16, FP32, and BF16.
19+
* **LLM Inference**: Supports inference for gpt2/llama/qwen models: [https://github.com/keith2018/TinyGPT](https://github.com/keith2018/TinyGPT)
2020

2121
## Implemented Operators and Components
2222

0 commit comments

Comments
 (0)