Tensors and Dynamic neural networks in Python with strong GPU acceleration
Repository dossier
tracel-ai/burn
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
Brief / README 摘要
查看原文*Burn is both a tensor library and a deep learning framework, optimized for numerical computing, training and inference.** Training and inference usually live in separate worlds. Models are typically trained in Python then exported to an open format like ONNX or optimized for production engines like vLLM, ONNX Runtime, or TensorRT. This export step is often brittle and lossy, ruling out complex architectures and advanced deployment use cases. Burn unifies the two. By executing multi-platform tensor operations via a single, unified API, the exact code used for training is the exact code that ru…
Topics / 标签
相关推荐
Implement a ChatGPT-like LLM in PyTorch from scratch, step by step
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
An Open Source Machine Learning Framework for Everyone
List of Computer Science courses with video lectures.