5 citations · 5 across the 2 of their papers we have counts for
3 papers · 1 filter
Generalisable Agents for Neural Network Optimisation
Kale-ab Tessera, Callum Rhys Tilbury, Sasha Abramowitz +5
Optimising deep neural networks is a challenging task due to complex training dynamics, high computational requirements, and long training times. To address this difficulty, we pro…
Mava: a research library for distributed multi-agent reinforcement learning in JAX
Ruan de Kock, Omayma Mahjoub, Sasha Abramowitz +5
Multi-agent reinforcement learning (MARL) research is inherently computationally expensive and it is often difficult to obtain a sufficient number of experiment samples to test hyp…
Keep the Gradients Flowing: Using Gradient Flow to Study Sparse Network Optimization
Kale-ab Tessera, Sara Hooker, Benjamin Rosman
Training sparse networks to converge to the same performance as dense neural architectures has proven to be elusive. Recent work suggests that initialization is the key. However, w…