⁠OSS of MAX Mojo APIs⁠

I would like to know how I can access the open-sourced parts of the (now deprecated) MAX Mojo API? There has been a commit 2 days ago which deleted these respective files it seems. Would love to learn how I can manually integrate the respective Graph/Engine parts in my Mojo 25.3 project. This would be a huge help!! :slight_smile:

I don’t have the bandwidth to break it out right now, but it’s probably a good idea to move it into a library.

From @joe in Discord –

Hey @TilliFe - they were already open sourced! They got removed in [docs] Fixes code output for KVCacheInputSymbols · modular/modular@34c0209 · GitHub (ignore the title of the commit and how Copybara was being used to no longer manage these files in the public repo). All of max.graph , max.driver , max.tensor , and their tests are available in that commit I mentioned above. You can see the full history with git log -- mojo/max/src/max/graph for example (edited)

I have a ticket for our team internally to write a user migration code for the tensor types, but that has not been started yet - sorry!

1 Like

just as a brief reference: NABLA now includes the open-sourced MAX API modules in its compiler module. This is still very experimental and needs a lot of cleanup, but it means that Nabla is independent from the deprecated MAX API from now on. I plan to make this module good enough/cleanup/maintain, for others to use as well.

2 Likes

nice!