burn

    Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

    Language: toml

    Author: Bozena De Vries (@bozenad)

    11 stars · 550 views

    Files

    • Dilate_2_7x7_Rect_BORDER_REFLECT101.png (image)
    • Dilate_2_7x7_Rect_BORDER_REPLICATE.png (image)
    • Dilate_2_3x5_Cross.png (image)
    • grid_sample.rs (rs)
    • conv.rs (rs)
    • attention.rs (rs)
    • mod.rs (rs)
    • unfold.rs (rs)
    • mod.rs (rs)
    • web-assembly.md (md)
    • examples.md (md)
    • distributed (txt)
    • distributed.rs (rs)
    • data (txt)
    • compare.rs (rs)
    • tensor.rs (rs)
    • mod.rs (rs)
    • config (txt)
    • codecov.yml (yml)
    • docs (txt)
    • katex-header.html (html)
    • burn-book (txt)
    • .prettierrc.json (json)
    • book.toml (toml)
    • src (txt)
    • custom-training-loop.md (md)
    • building-blocks (txt)
    • config.md (md)
    • backend.md (md)
    • tensor.md (md)
    • README.md (md)
    • module.md (md)
    • learner.md (md)
    • autodiff.md (md)
    • dataset.md (md)
    • metric.md (md)
    • record.md (md)
    • dataset.png (image)
    • onnx-import.md (md)
    • motivation.md (md)
    • basic-workflow (txt)
    • model.md (md)
    • backend.md (md)
    • training-output.png (image)
    • inference.md (md)
    • export_weights.py (py)
    • README.md (md)
    • training.md (md)
    • data.md (md)
    • getting-started.md (md)
    • performance (txt)
    • README.md (md)
    • good-practices (txt)
    • kernel-selection.md (md)
    • README.md (md)
    • asynchronous-execution.md (md)
    • kernel-fusion.md (md)
    • distributed-computing.md (md)
    • quantization.md (md)
    • complex_nested (txt)
    • SUMMARY.md (md)
    • distributed-computing.md (md)
    • models-and-pretrained-weights.md (md)
    • overview.md (md)
    • saving-and-loading.md (md)
    • advanced (txt)
    • README.md (md)
    • backend-extension (txt)
    • README.md (md)
    • custom-wgpu-kernel.md (md)
    • custom-cubecl-kernel.md (md)
    • no-std.md (md)
    • file_io.rs (rs)
    • benchmarks.toml (toml)
    • deny.toml (toml)
    • README.md (md)
    • crates (txt)
    • burn-store (txt)
    • MIGRATION.md (md)
    • safetensors-tests (txt)
    • tests (txt)
    • multi_layer (txt)
    • multi_layer.py (py)
    • mod.rs (rs)
    • test_mod.rs (rs)
    • src (txt)
    • lib.rs (rs)
    • Cargo.toml (toml)
    • helpers.rs (rs)
    • pytorch (txt)
    • pickle_reader.rs (rs)
    • README.md (md)
    • benches (txt)
    • generate_unified_models.py (py)
    • zero_copy_loading.rs (rs)
    • unified_loading.rs (rs)
    • unified_saving.rs (rs)
    • resnet18_loading.rs (rs)
    • download_resnet18.py (py)
    • examples (txt)
    • half_precision.rs (rs)
    • burnpack_inspect.rs (rs)
    • pytorch-tests (txt)
    • tests (txt)
    • boolean (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • enum_module (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • missing_module_field (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • key_remap (txt)
    • export_weights.py (py)
    • export_weights.py (py)
    • mod.rs (rs)
    • batch_norm (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • conv2d (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • integer (txt)
    • mod.rs (rs)
    • buffer (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • group_norm (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • conv_transpose2d (txt)
    • export_weights.py (py)
    • non_contiguous_indexes (txt)
    • store.rs (rs)
    • reader.rs (rs)
    • mod.rs (rs)
    • conv_transpose1d (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • top_level_key (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • key_remap_chained (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • export_weights.py (py)
    • mod.rs (rs)
    • embedding (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • conv1d (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • layer_norm (txt)
    • mod.rs (rs)
    • config (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • linear (txt)
    • export_weights.py (py)
    • mod.rs (rs)
    • test_mod.rs (rs)
    • export_weights.py (py)
    • mod.rs (rs)
    • src (txt)
    • lib.rs (rs)
    • Cargo.toml (toml)
    • src (txt)
    • lib.rs (rs)
    • collector.rs (rs)
    • keyremapper.rs (rs)
    • filter.rs (rs)
    • safetensors (txt)
    • store.rs (rs)
    • mod.rs (rs)
    • tests (txt)
    • mixed_datatypes.rs (rs)
    • direct_access.rs (rs)
    • round_trip.rs (rs)
    • adapter.rs (rs)
    • filtering.rs (rs)
    • pytorch_import.rs (rs)
    • mod.rs (rs)
    • multi_layer_verify.rs (rs)
    • error_handling.rs (rs)
    • metadata.rs (rs)
    • integration.rs (rs)
    • adapter.rs (rs)
    • burnpack (txt)
    • base.rs (rs)
    • writer.rs (rs)
    • store.rs (rs)
    • reader.rs (rs)
    • mod.rs (rs)
    • tests (txt)
    • edge_cases.rs (rs)
    • writer.rs (rs)
    • round_trip.rs (rs)
    • zero_copy.rs (rs)
    • store.rs (rs)
    • reader.rs (rs)
    • header.rs (rs)
    • mod.rs (rs)
    • alignment.rs (rs)
    • mod.rs (rs)
    • lazy_data.rs (rs)
    • tests (txt)
    • store (txt)
    • test_data (txt)
    • generate_enum_test.py (py)
    • mod.rs (rs)
    • mod.rs (rs)
    • reader (txt)
    • test_data (txt)
    • test_data.py (py)
    • create_tar_format.py (py)
    • create_legacy_with_offsets.py (py)
    • mod.rs (rs)
    • Dilate_1_5x5_Rect.png (image)
    • Dilate_2_7x7_Cross_BORDER_REPLICATE.png (image)
    • simple_legacy.py (py)
    • traits.rs (rs)
    • applier.rs (rs)
    • tensor_snapshot.rs (rs)
    • apply_result.rs (rs)
    • Cargo.toml (toml)
    • burn-vision (txt)
    • tests (txt)
    • morphology.rs (rs)
    • images (txt)
    • morphology (txt)
    • Dilate_2_7x7_Cross_BORDER_REFLECT.png (image)
    • Erode_1_5x5_Ellipse.png (image)
    • Dilate_2_5x7_Rect_ANCHOR.png (image)
    • Dilate_1_3x5_Cross.png (image)
    • Erode_1_5x5_Cross.png (image)
    • Dilate_2_7x7_Rect_BORDER_REFLECT.png (image)
    • Dilate_1_3x5_Rect.png (image)
    • Erode_1_5x5_Rect.png (image)
    • Dilate_2_7x7_Cross_BORDER_REFLECT101.png (image)
    • Base_1.png (image)
    • Dilate_1_5x5_Cross.png (image)
    • Base_2.png (image)
    • Dilate_2_5x7_Cross_ANCHOR_BORDER_REPLICATE.png (image)
    • Dilate_1_5x5_Ellipse.png (image)
    • Dilate_2_3x5_Rect.png (image)
    • nms.rs (rs)
    • common (txt)
    • mod.rs (rs)
    • connected_components.rs (rs)
    • src (txt)
    • lib.rs (rs)
    • utils (txt)
    • save.rs (rs)
    • mod.rs (rs)
    • base.rs (rs)
    • backends (txt)
    • mod.rs (rs)
    • cpu (txt)
    • base.rs (rs)
    • nms.rs (rs)
    • connected_components (txt)
    • spaghetti_4c (txt)
    • Spaghetti4C_forest_labels.rs (rs)
    • Spaghetti4C_first_line_forest_code.rs (rs)
    • mod.rs (rs)
    • Spaghetti4C_center_line_forest_code.rs (rs)
    • spaghetti (txt)
    • Spaghetti_forest_labels.rs (rs)
    • Spaghetti_first_line_forest_code.rs (rs)
    • Spaghetti_center_line_forest_code.rs (rs)
    • Spaghetti_single_line_forest_code.rs (rs)
    • mod.rs (rs)
    • Spaghetti_last_line_forest_code.rs (rs)
    • morphology (txt)
    • filter_engine.rs (rs)
    • filter.rs (rs)
    • mod.rs (rs)
    • connected_components.rs (rs)
    • mod.rs (rs)
    • ops.rs (rs)
    • cube (txt)
    • connected_components (txt)
    • prefix_sum.rs (rs)
    • hardware_accelerated.rs (rs)
    • mod.rs (rs)
    • mod.rs (rs)
    • ops.rs (rs)
    • tensor.rs (rs)
    • ops (txt)
    • base.rs (rs)
    • mod.rs (rs)
    • transform (txt)
    • transform2d.rs (rs)
    • mod.rs (rs)
    • loss (txt)
    • gram_matrix (txt)
    • gram_matrix_loss.rs (rs)
    • mod.rs (rs)
    • vgg19.rs (rs)
    • weights.rs (rs)
    • mod.rs (rs)
    • Cargo.toml (toml)
    • burn-backend (txt)
    • README.md (md)
    • src (txt)
    • lib.rs (rs)
    • backend (txt)
    • base.rs (rs)
    • primitive.rs (rs)
    • device.rs (rs)
    • ops (txt)
    • bool_tensor.rs (rs)
    • int_tensor.rs (rs)
    • qtensor.rs (rs)
    • modules (txt)
    • linear.rs (rs)
    • base.rs (rs)
    • ctc.rs (rs)
    • pool.rs (rs)
    • cat.rs (rs)
    • transaction.rs (rs)
    • tensor.rs (rs)
    • activation.rs (rs)
    • repeat_dim.rs (rs)
    • sort.rs (rs)
    • argwhere.rs (rs)
    • client.rs (rs)
    • server.rs (rs)
    • api.rs (rs)
    • mod.rs (rs)
    • ops.rs (rs)
    • mod.rs (rs)
    • alias.rs (rs)
    • cubecl.rs (rs)
    • quantization (txt)
    • scheme.rs (rs)
    • parameters.rs (rs)
    • mod.rs (rs)
    • Cargo.toml (toml)
    • burn-std (txt)
    • README.md (md)
    • tests (txt)
    • config.rs (rs)
    • src (txt)
    • lib.rs (rs)
    • element (txt)
    • cast.rs (rs)
    • base.rs (rs)
    • scalar.rs (rs)
    • mod.rs (rs)
    • distribution.rs (rs)
    • tensor (txt)
    • slice.rs (rs)
    • container.rs (rs)
    • dtype.rs (rs)
    • shape.rs (rs)
    • quantization.rs (rs)
    • mod.rs (rs)
    • device_settings.rs (rs)
    • ops.rs (rs)
    • id.rs (rs)
    • logger.rs (rs)
    • base.rs (rs)
    • fusion.rs (rs)
    • mod.rs (rs)
    • autodiff.rs (rs)
    • network.rs (rs)
    • Cargo.toml (toml)
    • burn-dispatch (txt)
    • README.md (md)
    • build.rs (rs)
    • src (txt)
    • remote_server.rs (rs)
    • lib.rs (rs)
    • device.rs (rs)
    • tensor.rs (rs)
    • ops (txt)
    • bool_tensor.rs (rs)
    • int_tensor.rs (rs)
    • qtensor.rs (rs)
    • module.rs (rs)
    • transaction.rs (rs)
    • tensor.rs (rs)
    • activation.rs (rs)
    • mod.rs (rs)
    • distributed.rs (rs)
    • backend.rs (rs)
    • macros.rs (rs)
    • Cargo.toml (toml)
    • burn-tch (txt)
    • README.md (md)
    • build.rs (rs)
    • src (txt)
    • lib.rs (rs)
    • cuda_hack (txt)
    • dummy_cuda_dependency.cpp (cpp)
    • fake_cuda_dependency.cpp (cpp)
    • element.rs (rs)
    • ATTRIBUTION.md (markdown)

    Loading code snippet…