…and DNA-sequence models in JAX or PyTorch, improves distributed training and inference, and develops research libraries and evaluation tools.
Job · Machine Learning
JAX
2 results
Upcoming first, then newest
Toward a biophysically-detailed, fully-differentiable model of the mouse retina
Kyra Kadhim, Ziwei Huang, Michael Deistler et al. · Bernstein Conference 2024
…Our model is based on the Jaxley software library (Deistler et al. 2024, Bernstein Conference) and JAX machine learning framework. The model consists of a network of Hodgkin-Huxley style neurons, realistic synapse dynamics, an experimentally-informed organization of cells in 3D space, and spatially constrained connectivity. We further implemented a wide v
ePoster · Computational Neuroscience