leroybird

leroybird's Stars

• kingoflolz/mesh-transformer-jax

  Model parallel transformers in JAX and Haiku

  Language:Python6.3k112206892

• google/flax

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

  Language:Jupyter Notebook6.1k86925647

• qtile/qtile

  :cookie: A full-featured, hackable tiling window manager written and configured in Python (X11 + Wayland)

  Language:Python4.8k962.1k702

• google-deepmind/optax

  Optax is a gradient processing and optimization library for JAX.

  Language:Python1.7k35254193

• lucidrains/lambda-networks

  Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute

  Language:Python1.5k4628157

• jessemelpolio/non-stationary_texture_syn

  Code used for texture synthesis using GAN

  Language:Python369161866

• jacobjinkelly/easy-neural-ode

  Code for the paper "Learning Differential Equations that are Easy to Solve"

  Language:Python27011831

• nestordemeure/flaxOptimizers

  A collection of optimizers, some arcane others well known, for Flax.

  Language:Python29501

• nestordemeure/AdaHessianJax

  Jax implementation of the AdaHessian optimizer

  Language:Python19513

• ai2cm/fv3core

  This repository has moved, please visit https://github.com/ai2cm/pace for the latest development of fv3core.

  Language:Python1310127

• sooheon/perceiver-jax

  Perceiver (transformer variant) implemented in JAX and Flax

  Language:Python11312

• sdatkinson/madgrad

  JAX implementation of MADGRAD

  Language:Python8301

• tethys-ts/tethysts

  Language:Python22202