georgedahl

Pinned Repositories

wip-constrained-extractor
Work in progress inference, learning, and evaluation code for extractive summarization.
Language:Python6 3 04
wip-lambada-lm
LSTM language model on LAMBADA dataset
Language:Python9 6 04
flax
Flax is a neural network library for JAX that is designed for flexibility.
Language:Python1 0 00
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:Python3 0 00
google-research
Google Research
Language:Jupyter Notebook32.8k 743 1.2k7.7k
tuning_playbook
A playbook for systematically maximizing the performance of deep learning models.
24.7k 270 342.1k
init2winit
Language:Python71 5 010
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:Python27.9k 320 5.1k2.6k
GPJax
Gaussian processes in JAX.
Language:Python385 4 16647
algorithmic-efficiency
MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvements in both training algorithms and models.
Language:Python297 22 20656

georgedahl's Repositories

georgedahl/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:Python3 0 00
georgedahl/flax
Flax is a neural network library for JAX that is designed for flexibility.
Language:Python1 0 00