Google DeepMind
Results
119
repositories owned by
Google DeepMind
kfac-jax
204
Stars
15
Forks
Watchers
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
conformal_training
108
Stars
7
Forks
Watchers
This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classifiers".
dks
52
Stars
2
Forks
Watchers
Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to tr...