# Pytorch Geometric in RLLib?

**URL:** <https://discuss.ray.io/t/pytorch-geometric-in-rllib/3079>\
**Category:** RLlib\
**Created:** [July 30, 2021, 7:27am UTC](https://discuss.ray.io/t/pytorch-geometric-in-rllib/3079 "2021-07-30T07:27:53Z")\
**Posts on this page:** 1\
**Showing post:** 3

<div class="post-metadata">

**Author:** ![smorad](https://sea2.discourse-cdn.com/flex020/user_avatar/discuss.ray.io/smorad/32/272_2.png) [@smorad](https://discuss.ray.io/u/smorad)\
**Post date:** [August 9, 2021, 12:27pm UTC](https://discuss.ray.io/t/pytorch-geometric-in-rllib/3079/3 "2021-08-09T12:27:17Z")

</div>

> [@Astariul](#):
>
> This library has a specific way to collate samples together in a batch (because each sample has a graph of different size, more info [here](https://pytorch-geometric.readthedocs.io/en/latest/notes/batching.html#)).

I use torch geometric in ray rllib here: [https://github.com/smorad/graph-conv-memory-paper/blob/master/src/models/ray\_graph.py](https://github.com/smorad/graph-conv-memory-paper/blob/master/src/models/ray_graph.py)

---

_[View the full topic](https://discuss.ray.io/t/pytorch-geometric-in-rllib/3079)._