bornaahz
- Karma
- 0
- Created
- ()
- Submissions
- 0
-
flaim is a library of state-of-the-art pre-trained vision models, plus common deep learning modules in computer vision, for JAX/Flax. It exposes a host of diverse image models through a straightforward interface with an…
-
open_clip_jax is an open source JAX/Flax implementation of OpenAI's CLIP, including image and text towers, pre-trained parameters, training utilities that can scale to TPU Pods, and more. It is inspired by but not…
-
minbpe-hs is a port of Andrej Karpathy's concise byte-level byte pair encoding (BPE) implementation, minbpe, to Haskell. During training, BPE compresses a sequence by repeatedly finding the most frequent pair of…
-
attorch is a subset of PyTorch's nn module, written purely in Python using OpenAI's Triton. Its goal is to be an easily hackable, self-contained, and readable collection of neural network modules whilst maintaining or…