4 citations · 4 across the 2 of their papers we have counts for
2 papers
cs.LG2023
Moccasin: Efficient Tensor Rematerialization for Neural Networks
Burak Bartan, Haoming Li, Harris Teague +2
The deployment and training of neural networks on edge computing devices pose many challenges. The low memory nature of edge devices is often one of the biggest limiting factors en…
cs.LG2022★ 4 cited
Neural Topological Ordering for Computation Graphs
Mukul Gagrani, Corrado Rainone, Yang Yang +7
Recent works on machine learning for combinatorial optimization have shown that learning based approaches can outperform heuristic methods in terms of speed and performance. In thi…