google-deepmind/dks
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 train.
No commits in the last 6 months.
Stars
76
Forks
5
Language
Python
License
Apache-2.0
Category
Last pushed
Jul 01, 2025
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/google-deepmind/dks"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
extropic-ai/thrml
Thermodynamic Hypergraphical Model Library in JAX