GitHub · Where software is built

Implement all Ops in PyTorch (help welcome!)
#821 · ricardoV94 opened on Jun 14, 2024
29
Speed up test times 🚀
#1124 · williambdean opened on Dec 13, 2024
4

Labels Milestones New issue

Allow vectorization of graphs with<code>IfElse</code>

#1510

· jessegrabowski opened

on Jul 1, 2025

<code>OpFromGraph</code> falls back to object mode when output shape graph contains <code>Blockwise</code>

shape inference

#1508

· jessegrabowski opened

on Jul 1, 2025

<code>numba_funcify_Elemwise</code> can end up with <code>parent_node</code> in its kwargs when vectorizing <code>OpFromGraph</code>

#1507

· jessegrabowski opened

on Jul 1, 2025

Remove BatchedDot and provide C implementation for batched matmul that uses numpy C-API

#1357

· ricardoV94 opened

on Apr 9, 2025

Inplace Blockwise outputs on inputs in Numba backend

Op implementation

#1030

· ricardoV94 opened

on Oct 11, 2024

Handle non-square intermediate Blockwise operations

#1017

· ricardoV94 opened

on Oct 6, 2024

Vectorize node should return list of variables not node

#902

· ricardoV94 opened

on Jul 8, 2024

Add option to implement vectorized hessian/jacobian

#756

· ricardoV94 opened

on May 8, 2024

Implement <code>lu_factor</code> and <code>lu_solve</code> for Blockwise <code>Solve</code> rewrites

backend compatibility

feature request

graph rewriting

SciPy compatibility

#582

· jessegrabowski opened

on Jan 10, 2024

Reconsider use of MakeSlice and NoneConst as inputs to AdvancedIndexing

Op implementation

#541

· ricardoV94 opened

on Dec 9, 2023