Tristan Bilot

icon indicating a website link https://tristanbilot.github.io

icon company Université Paris-Saclay icon location Paris, France icon location Ph.D. student | GNNs for Cybersecurity.

Results 3 issues of Tristan Bilot

Add Message Passing Neural Network (MPNN)

22 comment

## Proposed changes This PR proposes a draft for the [MPNN](https://proceedings.mlr.press/v70/gilmer17a/gilmer17a.pdf) architecture, used as a basic block to build almost any GNNs using message passing paradigm. This abstract architecture is...

[BUG] cumsum fails with int64 array

2 comment

**Describe the bug** When applying cumsum to an int64 array, there's a kernel error. **To Reproduce** ```python import mlx.core as mx a = mx.array([0, 1, 2], dtype=mx.int64) mx.cumsum(a) >>> libc++abi:...

enhancement

Alternative to nonzero()

5 comment

This function is still not implemented. * Do you know an efficient workaround until the feature is implemented? * Regarding the implementation, my first intuition is that we need a...

question