Code

Add groups to Conv1d#948

Code

Merged

awni merged 10 commits intoml-explore:mainml-explore/mlx:mainfrom Rifur13:groupsRifur13/mlx:groupsCopy head branch name to clipboardApr 27, 2024

+634-56

Commits

Commits on Apr 27, 2024

• Add conv1d grouped convs on CPU

  Rifur13

  committed
  8096ba6

• Add GPU support

  Rifur13

  committed
  ef666a1

• Parallelize inside metal kernel

  Rifur13

  committed
  029bf18

• clenaup

  Rifur13

  committed
  3463a9f

• Update mlx/ops.cpp

  Show description for 0787595
  

  Rifur13

  and

  awni

  committed
  0787595

• New unfold kernel + remove unused code

  Rifur13

  committed
  34382a3

• Remove copy and refactor

  Rifur13

  committed
  13499e9

• Update vjp and reuse steel gemm

  Rifur13

  committed
  2b43998

• Fixed groups on cpu

  Rifur13

  committed
  861ec7e

• Fix metal validation

  Rifur13

  committed
  48efb11