Skip to content
View Phimos's full-sized avatar
😶
😶
54 followers · 137 following

Achievements

Achievement: YOLOAchievement: Pull SharkAchievement: StarstruckAchievement: Arctic Code Vault Contributor

Achievements

Achievement: YOLOAchievement: Pull SharkAchievement: StarstruckAchievement: Arctic Code Vault Contributor

Block or report Phimos

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
Stars

jax

8 repositories

n2cholas / awesome-jax

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

1,994 155 Updated Sep 2, 2025

google / flax

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

Jupyter Notebook 6,986 770 Updated Dec 21, 2025

google-deepmind / dm-haiku

JAX-based neural network library

Python 3,146 271 Updated Dec 18, 2025

google / trax

Trax — Deep Learning with Clear Code and Speed

Python 8,296 827 Updated Sep 26, 2025

poets-ai / elegy

A High Level API for Deep Learning in JAX

Python 476 31 Updated Dec 15, 2022

kingoflolz / mesh-transformer-jax

Model parallel transformers in JAX and Haiku

Python 6,356 889 Updated Jan 21, 2023

metaopt / optree

OpTree: Optimized PyTree Utilities

Python 202 12 Updated Dec 21, 2025

google-deepmind / penzai

A JAX research toolkit for building, editing, and visualizing neural networks.

Python 1,830 68 Updated Jun 22, 2025