Code

MLX backend POC#1365

Code

Open

williambdean wants to merge 56 commits intopymc-devs:mainpymc-devs/pytensor:mainfrom

williambdean:mlx-pocwilliambdean/pytensor:mlx-pocCopy head branch name to clipboard

Commits

Commits on Apr 11, 2025

• mlx poc

  williambdean

  committed
  d25f214

  Copy full SHA for d25f214

• add test for dot

  williambdean

  committed
  edacc0e

  Copy full SHA for edacc0e

• restore pytorch

  williambdean

  committed
  052fdc2

  Copy full SHA for 052fdc2

• wrap in mx.array

  williambdean

  committed
  a9ecad0

  Copy full SHA for a9ecad0

• modify the pytorch jit

  williambdean

  committed
  e690bff

  Copy full SHA for e690bff

• move file

  williambdean

  committed
  ad29c17

  Copy full SHA for ad29c17

• dont wrap

  williambdean

  committed
  ba29b37

  Copy full SHA for ba29b37

• attempt to fix github action

  williambdean

  committed
  8716870

  Copy full SHA for 8716870

• change the rtol

  williambdean

  committed
  9bf7edf

  Copy full SHA for 9bf7edf

• add init file

  williambdean

  committed
  96ba116

  Copy full SHA for 96ba116

• skip if not installed

  williambdean

  committed
  e116fa1

  Copy full SHA for e116fa1

• remove torch related code / comments

  williambdean

  committed
  5d5f754

  Copy full SHA for 5d5f754

Commits on Apr 12, 2025

• simplify the fgraph_convert

  williambdean

  committed
  b8cee3f

  Copy full SHA for b8cee3f

• assert type

  williambdean

  committed
  d057453

  Copy full SHA for d057453

Commits on Apr 18, 2025

• simplify the internal

  williambdean

  committed
  ae202e6

  Copy full SHA for ae202e6

• remove the language

  williambdean

  committed
  f1941fe

  Copy full SHA for f1941fe

• Adding operations in pytensor

  cetagostini

  committed
  7c8eae7

  Copy full SHA for 7c8eae7

• add extension

  williambdean

  committed
  67a74fb

  Copy full SHA for 67a74fb

• make compare function

  williambdean

  committed
  fb5eb52

  Copy full SHA for fb5eb52

• rename function

  williambdean

  committed
  516b595

  Copy full SHA for 516b595

• correct the function name

  williambdean

  committed
  67bb8da

  Copy full SHA for 67bb8da

• tests for elemwise

  williambdean

  committed
  82bb964

  Copy full SHA for 82bb964

• Changes

  cetagostini

  committed
  877d79f

  Copy full SHA for 877d79f

• Toma tu tomate William

  cetagostini

  committed
  fafedd6

  Copy full SHA for fafedd6

• Pushing changes with the core shit.

  cetagostini

  committed
  60acb8d

  Copy full SHA for 60acb8d

• add more tests

  williambdean

  committed
  242aba7

  Copy full SHA for 242aba7

• additional tests

  williambdean

  committed
  6cb47fc

  Copy full SHA for 6cb47fc

• test for switch with mlx

  williambdean

  committed
  bc98e09

  Copy full SHA for bc98e09

• Pushing code

  cetagostini

  committed
  4d5b34b

  Copy full SHA for 4d5b34b

• Changes

  cetagostini

  committed
  5abd32d

  Copy full SHA for 5abd32d

• A lot of new code

  cetagostini

  committed
  12daeac

  Copy full SHA for 12daeac

• almost there baby william

  cetagostini

  committed
  ac93949

  Copy full SHA for ac93949

• Another push small

  cetagostini

  committed
  a19cbc8

  Copy full SHA for a19cbc8

• fix for all

  williambdean

  committed
  5c97bc8

  Copy full SHA for 5c97bc8

• fix for carlos

  williambdean

  committed
  2fc81bc

  Copy full SHA for 2fc81bc

Commits on Apr 19, 2025

• just return the compiled func

  williambdean

  committed
  e6437cc

  Copy full SHA for e6437cc

• A change for willy may!

  cetagostini

  committed
  c3a3e1a

  Copy full SHA for c3a3e1a

• FINALLY BABY LETS PARTY! (IF YOU ARE READING THIS MAKE MORE PRs)

  cetagostini

  committed
  e7cf10e

  Copy full SHA for e7cf10e

• refactor to use getattr

  williambdean

  committed
  880dd5c

  Copy full SHA for 880dd5c

• bring argmax test

  williambdean

  committed
  1e6addd

  Copy full SHA for 1e6addd

• use deepcopy

  williambdean

  committed
  aabbb78

  Copy full SHA for aabbb78

• move some tests

  williambdean

  committed
  0812c55

  Copy full SHA for 0812c55

• THE SUPER BLOCKWISEE YA YA YA YA JUUUUU

  cetagostini

  committed
  294c271

  Copy full SHA for 294c271

• Merge branch 'mlx-poc' of https://github.com/williambdean/pytensor into pr/1365

  cetagostini

  committed
  9d3eca8

  Copy full SHA for 9d3eca8

• Guys, I'm getting sad. We need help yisus!!!!!

  cetagostini

  committed
  9f31ab1

  Copy full SHA for 9f31ab1

• WILLIAM YOU NEED TO GO ANOTHER MILE! GO ON MY MATEEEEEEE, GO PHILLIES!

  cetagostini

  committed
  37440ff

  Copy full SHA for 37440ff

• RETURN, WHAT A SHAME! Sad times are coming.

  cetagostini

  committed
  4e4923f

  Copy full SHA for 4e4923f

• AI COULD BE COOL? OR WE ARE JUST FUCKING AROUND?

  cetagostini

  committed
  6b27dc4

  Copy full SHA for 6b27dc4

• AI RULES BABY MY MATE

  cetagostini

  committed
  e308f83

  Copy full SHA for e308f83

• test conv1d case

  williambdean

  committed
  3744a18

  Copy full SHA for 3744a18

• I'm going for pizzas, it was an incredible day!

  cetagostini

  committed
  b41cab0

  Copy full SHA for b41cab0

• Merge branch 'mlx-poc' of https://github.com/williambdean/pytensor into pr/1365

  cetagostini

  committed
  323fa9d

  Copy full SHA for 323fa9d

• SUUUUUUUUU!!!!!! LIFE IS GOING WELL. MLX FOR MEDIA MIX MODELS BAY

  Show description for 9766975
  

  3 peoplecommitted
  9766975

  Copy full SHA for 9766975

• pre-commit

  cetagostini

  committed
  5ffc5ef

  Copy full SHA for 5ffc5ef

• Almost working

  cetagostini

  committed
  597f84e

  Copy full SHA for 597f84e

Commits on Apr 23, 2025

• Last PR sampling working

  Show description for fb8fd2f
  cetagostini

  committed
  fb8fd2f

  Copy full SHA for fb8fd2f