Skip to content

Conversation

@aseembits93
Copy link
Contributor

@aseembits93 aseembits93 commented Dec 30, 2025

WIP -> Support Numba, Pytorch, JAX, Tensorflow

MLX for future support (they do lazy evaluation so we have to do mlx.eval(func,args) instead of func(args))

@CLAassistant
Copy link

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.


Codeflash Bot seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
You have signed the CLA already but the status is still pending? Let us recheck it.

args=ast.arguments(
args=[], vararg=None, kwarg=None, posonlyargs=[], kwonlyargs=[], kw_defaults=[], defaults=[]
),
body=[pytorch_sync, jax_sync, mlx_sync, tensorflow_sync],
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

conditionally contain only one of them

@aseembits93
Copy link
Contributor Author

closing this in favor of smaller more focused Pull Requests @misrasaurabh1 @KRRT7

@aseembits93
Copy link
Contributor Author

not the correct branch for this PR, opening it with the correct one

@aseembits93 aseembits93 closed this Jan 9, 2026
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

Successfully merging this pull request may close these issues.

3 participants