Robert Dyro

@rdyro

JAX @ Google Mountain View, CA

Followers

Following

Public Repos

Private Repos

Language Breakdown

Lines of code distribution across 40 owned repositories

15.3M Total LOC

Jupyter Notebook

8,839,373 lines

57.7%

N/A

Python

4,882,520 lines

31.9%

N/A

JavaScript

455,780 lines

3.0%

N/A

HTML

389,911 lines

2.5%

N/A

C++

319,786 lines

2.1%

N/A

Other

422,640 lines

2.8%

N/A

T-Shaped Developer

T-shaped

Deep in Jupyter Notebook with broad versatility

Jupyter Notebook

Python

JavaScript

HTML

C++

Collaboration Network

Global Impact visualization

LIVE

0 active collaborators

Repos

PRs

Growth

+18%

Top Collaborators

No collaborator data yet.

Coding Streak

Contribution activity over the past year

1 day

423

Contributions

308

Commits

Pull Requests

Jun Jul Aug Sep Oct Nov Dec Jan Feb Mar Apr May Jun

Based on GitHub activity

Less

Followers 63

raikou

@happychow

Bill Sun

@QingyunSun

Glenn Gertler

@glenn-gertler

Sam Buchanan

@sdbuch

viralvgupta

@viralvgupta

View All

Following

0 total

Not following anyone yet.

Synced via GitHub

Top Repositories

torch2jax

Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.

59 3

Python

mpcjax

A JAX library for Sequential Convex Programming (SCP) Model Predictive Control (MPC) path planning

Microbenchmarking hyperparameter tuning for JAX functions.

18 0

Python

SpAutoDiff.jl

Full [Sp]arse Graph [AutoDiff]erentiation Package for Julia

5 1

Julia

jaxfi-JAXFriendlyInterface

Friendly Interface to JAX, `jaxfi` simplifies JAX interface to replicate PyTorch's ease of use.

Ragged dot op with auto-diff in JAX Pallas

Open Source Impact

Contributions to external projects

85 merged PRs

google-deepmind/optax

2284

conda-forge/optax-feedstock

AI-Hypercomputer/maxtext

2320

jax-ml/jax-llm-examples

256

openxla/tokamax

227

Contributed to 11 repositories