Viewing profile — bornaahz

Thanks a ton! It means a lot.

Hello everyone, I am the author of this project. If anyone has any questions concerning trap, I'd be more than happy to address them.

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 thro…

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, an…

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 fi…

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 …