Skip to content
View cpgaffney1's full-sized avatar

Organizations

@googlers

Block or report cpgaffney1

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Maximum 250 characters. Please don't include any personal information such as legal names or email addresses. Markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
Showing results

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Python 33,928 3,234 Updated Nov 9, 2025

Flax is a neural network library for JAX that is designed for flexibility.

Jupyter Notebook 6,900 755 Updated Nov 8, 2025

A simple, performant and scalable Jax LLM!

Python 1,973 420 Updated Nov 8, 2025

Orbax provides common checkpointing and persistence utilities for JAX users

Python 447 71 Updated Nov 8, 2025