jkyl

research engineer

MirageNew York City

Pinned Repositories

orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Language:Python426 8 13967
arxiv-bookmark-renamer
Chrome extension to rename bookmarked arXiv links to their paper titles
Language:JavaScript00
biggan
BigGAN in idiomatic Keras
Language:Python12 2 112
nnx-gru
A JAX / NNX implementation of a linear GRU for language modeling
Language:Jupyter Notebook0 1 00
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:C++0 0 00
tensorflow
An Open Source Machine Learning Framework for Everyone
Language:C++0 0 00
vq-vae
A JAX / NNX implementation of a VQ-VAE for audio compression
Language:Jupyter Notebook0 1 00
pytorch-fid
Compute FID scores with PyTorch.
Language:Python3.7k 12 90528
tensorflow
An Open Source Machine Learning Framework for Everyone
Language:C++192k 7.5k 40.8k74.8k

jkyl's Repositories

jkyl/biggan
BigGAN in idiomatic Keras
Language:Python12 2 112
jkyl/arxiv-bookmark-renamer
Chrome extension to rename bookmarked arXiv links to their paper titles
Language:JavaScript00
jkyl/nnx-gru
A JAX / NNX implementation of a linear GRU for language modeling
Language:Jupyter Notebook0 1 00
jkyl/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:C++0 0 00
jkyl/tensorflow
An Open Source Machine Learning Framework for Everyone
Language:C++0 0 00
jkyl/vq-vae
A JAX / NNX implementation of a VQ-VAE for audio compression
Language:Jupyter Notebook0 1 00