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.

Please don't include any personal information such as legal names or email addresses. Maximum 100 characters, 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 31,080 2,874 Updated Jan 27, 2025

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

Jupyter Notebook 6,290 660 Updated Jan 26, 2025

A simple, performant and scalable Jax LLM!

Python 1,597 309 Updated Jan 27, 2025

Orbax provides common checkpointing and persistence utilities for JAX users

Python 328 40 Updated Jan 27, 2025