{"id":21760071,"url":"https://github.com/andreytkachenko/yarnn","last_synced_at":"2025-07-05T08:02:04.151Z","repository":{"id":57672673,"uuid":"171877027","full_name":"andreytkachenko/yarnn","owner":"andreytkachenko","description":"Yet Another rust Neural Network framework","archived":false,"fork":false,"pushed_at":"2022-03-17T18:20:55.000Z","size":41787,"stargazers_count":14,"open_issues_count":0,"forks_count":0,"subscribers_count":2,"default_branch":"master","last_synced_at":"2024-04-25T21:43:43.485Z","etag":null,"topics":["convolutional-neural-networks","framework","machine-learning","neural-networks","rust"],"latest_commit_sha":null,"homepage":"","language":"Rust","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/andreytkachenko.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null}},"created_at":"2019-02-21T13:29:38.000Z","updated_at":"2023-10-18T15:38:40.000Z","dependencies_parsed_at":"2022-08-31T05:40:11.791Z","dependency_job_id":null,"html_url":"https://github.com/andreytkachenko/yarnn","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/andreytkachenko%2Fyarnn","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/andreytkachenko%2Fyarnn/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/andreytkachenko%2Fyarnn/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/andreytkachenko%2Fyarnn/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/andreytkachenko","download_url":"https://codeload.github.com/andreytkachenko/yarnn/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":248716892,"owners_count":21150387,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["convolutional-neural-networks","framework","machine-learning","neural-networks","rust"],"created_at":"2024-11-26T11:37:23.815Z","updated_at":"2025-04-13T12:41:06.294Z","avatar_url":"https://github.com/andreytkachenko.png","language":"Rust","funding_links":[],"categories":[],"sub_categories":[],"readme":"## Yet Another Rust Neural Network framework aka YARNN\nInspired by `darknet` and `leaf`\n\n## What it can right now:\n * not requires `std` (only `alloc` for tensor allocations, bump allocator is ok, so it can be compiled to stm32f4 board)\n * available layers: `Linear`, `ReLu`, `Sigmoid`, `Softmax`(no backward), `Conv2d`, `ZeroPadding2d`, `MaxPool2d`, `AvgPool2d`(no backward), `Flatten`\n * available optimizers: `Sgd`, `Adam`, `RMSProp`\n * available losses: `CrossEntropy`(no forward), `MeanSquareError`\n * available backends: `Native`, `NativeBlas`(no convolution yet)\n\n## What it will can (I hope):\n### 1st stage:\n * example of running `yarnn` in browser using `WASM`\n * example of running `yarnn` on `stm32f4` board\n * finish `AvgPool2d` backpropogation\n * add `Dropout` layer\n * add `BatchNorm` layer\n * convolution with BLAS support \n### 2nd stage:\n * `CUDA` support\n * `OpenCL` support\n### 3rd stage:\n * `DepthwiseConv2d` layer\n * `Conv3d` layer\n * `Deconv2d` layer\n * `k210` backend\n\n## Model definition example\n```rust\nuse yarnn::model;\nuse yarnn::layer::*;\nuse yarnn::layers::*;\n\nmodel! {\n    MnistConvModel (h: u32, w: u32, c: u32) {\n        input_shape: (c, h, w),\n        layers: {\n            Conv2d\u003cN, B, O\u003e {\n                filters: 8\n            },\n            ReLu\u003cN, B\u003e,\n            MaxPool2d\u003cN, B\u003e {\n                pool: (2, 2)\n            },\n\n            Conv2d\u003cN, B, O\u003e {\n                filters: 8\n            },\n            ReLu\u003cN, B\u003e,\n            MaxPool2d\u003cN, B\u003e {\n                pool: (2, 2)\n            },\n\n            Flatten\u003cN, B\u003e,\n            Linear\u003cN, B, O\u003e {\n                units: 10\n            },\n\n            Sigmoid\u003cN, B\u003e\n        }\n    }\n}\n```\n\n## Contributors are welcome","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fandreytkachenko%2Fyarnn","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fandreytkachenko%2Fyarnn","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fandreytkachenko%2Fyarnn/lists"}