burn
Deep learning framework
A comprehensive dynamic Deep Learning Framework built with Rust for extreme flexibility, compute efficiency, and portability.
Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals.
9k stars
65 watching
457 forks
Language: Rust
last commit: 3 months ago
Linked from 2 awesome lists
autodiffautotuneconcurrencycross-platformdeep-learninghigh-performancekernel-fusionllmmachine-learningndarrayneural-networkonnxpytorchrustscientific-computingtensorwasmwebgpu
josephmisiti/awesome-machine-learning
rust-unofficial/awesome-rustRelated projects:
| Repository | Description | Stars |
|---|---|---|
 weld-project/weld | Optimizes data-intensive applications by expressing core computations in a common intermediate representation and optimizing across multiple libraries and frameworks. | 2,996 |
 catalyst-team/catalyst | A PyTorch framework for accelerating deep learning research and development by focusing on reproducibility, rapid experimentation, and codebase reuse. | 3,300 |
 pytorch/glow | A compiler and execution engine for neural networks that generates optimized code for hardware accelerators | 3,247 |
 deepseek-ai/deepseek-coder | A code completion model trained on large amounts of programming language data to help developers write code more efficiently. | 6,987 |
 donnyyou/torchcv | A comprehensive PyTorch-based framework for computer vision tasks | 2,249 |
 autumnai/leaf | An open machine learning framework for building classical, deep, or hybrid models on various hardware platforms. | 5,555 |
 bvlc/caffe | A framework for building and training deep learning models, particularly those using convolutional neural networks. | 34,149 |
 coreylowman/dfdx | A deep learning library for Rust with GPU acceleration and ergonomic API. | 1,754 |
 nvidia/digits | A web-based platform for training deep learning models on GPU accelerators. | 4,126 |
 apache/tvm | Compiler stack for deep learning systems to improve performance and efficiency on CPU, GPU, and specialized hardware. | 11,853 |
 flame/blis | A software framework for instantiating high-performance linear algebra libraries | 2,337 |
 weiliu89/caffe | A deep learning framework for rapid prototyping and production of computer vision applications | 4,770 |
 deepjavalibrary/djl | A high-level Java framework for building and deploying deep learning models | 4,204 |
 guillaume-be/rust-bert | Provides pre-trained Rust-native transformer-based models and pipelines for Natural Language Processing tasks. | 2,694 |
 plasma-umass/coz | An open-source profiler that uses causal profiling to measure optimization potential and predict the impact of code optimizations on performance. | 4,136 |