Code

Fast argmax#66

Code

Open

AnihilatorGun wants to merge 22 commits intomasterneuro-ml/imops:masterfrom argmaxneuro-ml/imops:argmaxCopy head branch name to clipboard

+296-6

Commits

Commits on Jan 29, 2025

• initial argmax

  AnihilatorGun

  committed
  b64cae5

  Copy full SHA for b64cae5

Commits on Jan 30, 2025

• fix argmax, build, authors:)

  AnihilatorGun

  committed
  4837074

  Copy full SHA for 4837074

• version

  AnihilatorGun

  committed
  a722bb2

  Copy full SHA for a722bb2

• argmax for different positions

  AnihilatorGun

  committed
  0912a6d

  Copy full SHA for 0912a6d

• argmax as python func

  AnihilatorGun

  committed
  1fb32fd

  Copy full SHA for 1fb32fd

• docs

  AnihilatorGun

  committed
  b5e1b19

  Copy full SHA for b5e1b19

• speed up when pre_dim is small

  AnihilatorGun

  committed
  ac1bdfe

  Copy full SHA for ac1bdfe

• flake8

  AnihilatorGun

  committed
  c138175

  Copy full SHA for c138175

• march=native

  AnihilatorGun

  committed
  74f5ea5

  Copy full SHA for 74f5ea5

• O3 optimizations

  AnihilatorGun

  committed
  bb5893d

  Copy full SHA for bb5893d

• Fallback to numpy

  AnihilatorGun

  committed
  9952ec2

  Copy full SHA for 9952ec2

• limit num_threads, test_argmax

  AnihilatorGun

  committed
  f98beab

  Copy full SHA for f98beab

• flake8

  AnihilatorGun

  committed
  1ca4cc1

  Copy full SHA for 1ca4cc1

Commits on Feb 3, 2025

• style, no backend passed in argmax

  AnihilatorGun

  committed
  7c78f7c

  Copy full SHA for 7c78f7c

• test

  AnihilatorGun

  committed
  5f3d639

  Copy full SHA for 5f3d639

• warn if input is not contiguous

  AnihilatorGun

  committed
  aaea5f6

  Copy full SHA for aaea5f6

• flake8

  AnihilatorGun

  committed
  ee6b4a5

  Copy full SHA for ee6b4a5

• black setup

  AnihilatorGun

  committed
  5013a18

  Copy full SHA for 5013a18

• variables naming

  AnihilatorGun

  committed
  4daab36

  Copy full SHA for 4daab36

• pyx style

  AnihilatorGun

  committed
  163a06f

  Copy full SHA for 163a06f

Commits on Feb 4, 2025

• test

  AnihilatorGun

  committed
  9927037

  Copy full SHA for 9927037

• comment

  AnihilatorGun

  committed
  45cce6c

  Copy full SHA for 45cce6c