All pull requests

New pull request

Implement grad function and enable SUPPORTS_GRADIENT
#44PlumBlossomMaid opened Oct 2, 2026Collaborator
Support order>1 in map_coordinates via numpy fallback
#43PlumBlossomMaid opened Oct 2, 2026Collaborator
Support even kernel sizes in gaussian_blur
#42PlumBlossomMaid opened Oct 2, 2026Collaborator
Add grad function (raises NotImplementedError)
#41PlumBlossomMaid opened Oct 2, 2026Collaborator
Support all numpy.pad modes via numpy fallback
#40PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement synchronized=True in moments
#39PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement attn_logits_soft_cap in dot_product_attention
#38PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement reconstruct_patches_3d op
#37PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement extract_patches_3d op
#36PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement ssim op
#35PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement reconstruct_patches op
#34PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement extract_patches op
#33PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement pad_images and crop_images ops
#32PlumBlossomMaid opened Oct 2, 2026Collaborator
Add SUPPORTS_GRADIENT flag
#31PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement view_as_real and view_as_complex ops
#30PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement polar op
#29PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement layer_normalization op
#28PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement rms_normalization op
#27PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement normalize op
#26PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement saturate_cast op
#25PlumBlossomMaid opened Oct 2, 2026Collaborator
Implement dtype op
#24PlumBlossomMaid opened Oct 2, 2026Collaborator
Add SUPPORTS_GRADIENT flag, exclude more tests.
#23hertschuh opened Oct 2, 2026Collaborator
Implement affine_transform with correct reflect index mapping
#16PlumBlossomMaid opened Sep 27, 2026Collaborator
3