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.

GitHub

9k stars
67 watching
445 forks
Language: Rust
last commit: 3 days ago
Linked from 2 awesome lists

autodiffautotuneconcurrencycross-platformdeep-learninghigh-performancekernel-fusionllmmachine-learningndarrayneural-networkonnxpytorchrustscientific-computingtensorwasmwebgpu

Backlinks from these awesome lists:

Related 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,995
catalyst-team/catalyst A PyTorch framework for accelerating deep learning research and development by focusing on reproducibility, rapid experimentation, and codebase reuse. 3,295
pytorch/glow A compiler and execution engine for neural networks that generates optimized code for hardware accelerators 3,235
deepseek-ai/deepseek-coder A code completion model trained on large amounts of programming language data to help developers write code more efficiently. 6,837
donnyyou/torchcv A comprehensive PyTorch-based framework for computer vision tasks 2,250
autumnai/leaf An open machine learning framework for building classical, deep, or hybrid models on various hardware platforms. 5,558
bvlc/caffe A framework for building and training deep learning models, particularly those using convolutional neural networks. 34,125
coreylowman/dfdx A deep learning library for Rust with GPU acceleration and ergonomic API. 1,737
nvidia/digits A web-based platform for training deep learning models on GPUs using Caffe, Torch, and other frameworks. 4,121
apache/tvm Compiler stack for deep learning systems to improve performance and efficiency on CPU, GPU, and specialized hardware. 11,788
flame/blis A software framework for instantiating high-performance linear algebra libraries 2,312
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,144
guillaume-be/rust-bert Provides pre-trained Rust-native transformer-based models and pipelines for Natural Language Processing tasks. 2,651
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,106