jysohn23

Senior Software Engineer @ Google Brain

Google DeepMind

Pinned Repositories

paxml
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
Language:Python463 15 2169
praxis
Language:Python180 8 1242
CapsuleNetwork
Implementation of Dynamic Routing Between Capsules
Language:Python2 2 30
transformers
🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
Language:Python1 1 00
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
Language:C++2.5k 58 2.4k487

jysohn23/CapsuleNetwork
Implementation of Dynamic Routing Between Capsules
Language:Python2 2 30
jysohn23/transformers
🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
Language:Python1 1 00
jysohn23/blog
Public repo for HF blog posts
Language:Jupyter Notebook0 0
jysohn23/examples
This repository contains example code to build models on TPUs
Language:Shell0 0
jysohn23/fairseq
Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
Language:Python0 0
jysohn23/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:Python0 0
jysohn23/maskrcnn-benchmark
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
Language:Python1 0
jysohn23/models
Models and examples built with TensorFlow
Language:Python0 0
jysohn23/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
jysohn23/vision
Datasets, Transforms and Models specific to Computer Vision
Language:Jupyter Notebook
jysohn23/xla
Language:C++0 0