xla
[Kernel] Use call_jax to simplify the gmm pallas kernel wrapper
#9180

Merged

[Kernel] Use call_jax to simplify the gmm pallas kernel wrapper #9180

yaochengji merged 2 commits into master from chengji/gmm-use-call_jax

bhavya01 requested a review from

miladm 308 days ago

yaochengji requested a review from

tengyifei 308 days ago

yaochengji requested a review from

qihqi 308 days ago

yaochengji requested a review from

lsy323 308 days ago

yaochengji requested a review from

vanbasten23 308 days ago

vanbasten23 commented on 2025-05-16

[Kernel] Use call_jax to simplify the gmm pallas kernel wrapper

27a4c0a5

yaochengji force pushed from b7965145 to 27a4c0a5 308 days ago

Merge branch 'master' into chengji/gmm-use-call_jax

2d07b534

vanbasten23 commented on 2025-05-16

vanbasten23 approved these changes on 2025-05-16

yaochengji merged f39434ab into master 307 days ago

Reviewers

vanbasten23

miladm

tengyifei

qihqi

lsy323

Assignees

No one assigned

Labels

None yet

Milestone

No milestone