tracel-ai/burn

Linear algebra operations

Open

#1,538 opened on Mar 27, 2024

View on GitHub
 (14 comments) (14 reactions) (0 assignees)Rust (964 forks)auto 404
enhancementfeaturehelp wanted

Repository metrics

Stars
 (15,550 stars)
PR merge metrics
 (PR metrics pending)

Description

Feature description

Add common linear algebra operations.

Feature motivation

We've had a couple of users ask about support for linalg ops like SVD, QR decomposition and matrix exponential. We added matrix operations like the diagonal but don't have matrix/vector norm or determinant.

Suggest a Solution

Maybe in a new burn-linalg crate, but would have to see how that would work for support on different backends.

Contributor guide

Research direction
Examine the existing matrix operations in Burn, particularly the diagonal implementation and backend trait patterns. Identify where linear algebra operations like SVD, QR, determinant, and norm should be added. Evaluate the feasibility of a separate burn linalg crate and how it would interact with different backends. Start by implementing a simple norm function to understand the process.
Tech stack
rust
Domain
aibackend
Issue type
Feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
4
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
3-5 days
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Active
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Clear
Prerequisites
Rust
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
40