Skip to content
View frazane's full-sized avatar
29 followers · 83 following

Achievements

Achievement: StarstruckAchievement: Pair ExtraordinaireAchievement: Pull Sharkx2Achievement: YOLO

Achievements

Achievement: StarstruckAchievement: Pair ExtraordinaireAchievement: Pull Sharkx2Achievement: YOLO
Block or Report

Block or report frazane

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
Stars

jax

7 repositories

google / flax

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

Python 5,800 613 Updated Jul 9, 2024

n2cholas / awesome-jax

JAX - A curated list of resources https://github.com/google/jax

1,380 117 Updated Jul 8, 2024

google / jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

Python 903 62 Updated May 17, 2024

patrick-kidger / jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

Python 1,037 49 Updated Jul 2, 2024

google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.

Python 1,557 164 Updated Jul 9, 2024

dfm / extending-jax

Extending JAX with custom C++ and CUDA code

Python 361 21 Updated Nov 3, 2023

google / grain

Python 216 16 Updated Jul 5, 2024