Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

could it be made standalone ? #3

Open
nestordemeure opened this issue Mar 26, 2021 · 2 comments
Open

could it be made standalone ? #3

nestordemeure opened this issue Mar 26, 2021 · 2 comments

Comments

@nestordemeure
Copy link

Hello, I just discovered the project (great work by the way) and I wondered if it would be possible to turn it into a library.

My unsderstanding is that Tensat taxes a ONNX files, optimizes it (using TASO to get a cost function, store information, etc) and outputs an optimized ONNX.

Would it be possible to removes its dependency on TASO (using a Rust ONNX runtime for the evaluation part I guess) so that Tensat could be easily used as a library taking ONNX files (and maybe an AST as an alternative) and returning optimized versions that could then be run ?

I am asking the question because I am wondering about the feasability of building a JAX like library in Rust : using something akin to ndArray to produce a computational graph that would then be optimized (which is where Tensat could come into play) and compiled to GPU.

@remysucre
Copy link
Member

That's a great idea, and yes I think it's possible. FYI the contributors to this project are all grad students, and we have moved on to focus on other projects, so it's unlikely we'll make significant changes to tensat in the future. Feel free to fork and we welcome PR's.

@nestordemeure
Copy link
Author

Ok! I do not have the time to do it in the upcoming weeks but I am writing that in my long term TODO list...

I am keeping this issue open: if anyone wants to do it, do not hesitate to comment here!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants