google-deepmind/kfac-jax
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
PythonApache-2.0
Stargazers
- A-Alaa
- aleximmerETH Zurich, Max Planck Institute for Intelligent Systems
- chaeyeunparkXanadu
- chester-tan
- dandelin@naver-ai
- dsx-aishanghai
- dvtailorAmsterdam, Netherlands
- f-dangel@ProbabilisticNumerics @VectorInstitute
- fly51flyPRIS
- gd-zhangUniversity of Toronto
- georgedeathUniversity of Exeter
- GiantElephant123ByteDance
- gurusuraSura Systems Private Limited
- Harry-HopkinsonEconsumer
- hbq1London, UK
- JacobFV@Limboid @ComputaCo
- jacobjinkelly@deepmind
- jmbrSan Francisco Bay Area, California
- joaogui1
- joeryjoeryTU Delft
- Leticia-maria@JuliaChemicalReactions, @HartreeFoca
- lxuechenStanford University
- makdoudNPalaiseau, France
- mtkwThttps://revisio.co.jp/
- nikitavoloboevTbilisi
- ntselepidis@NVIDIA, ETH Zurich
- odellus@phytomech
- pharringtonp19
- pomonam@VectorInstitute
- soraros
- subhobrata
- taesiriPlanet Mars
- wangleiphyInstitute of Physics, CAS
- wilcolnGoogle DeepMind
- Yupei-DuUtrecht University
- zeusAlgo