sayakpaul / BiT-jax2tf
This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
☆14Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for BiT-jax2tf
- Shows how to do parameter ensembling using differential evolution.☆10Updated 2 years ago
- A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis.☆13Updated 2 years ago
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Updated 3 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated last year
- reproduces experiments from "Grounding inductive biases in natural images: invariance stems from variations in data"☆16Updated last month
- Contains Colab Notebooks show cool use-cases of different GCP ML APIs.☆10Updated 4 years ago
- Local Attention - Flax module for Jax☆20Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 3 years ago
- ☆24Updated 2 years ago
- Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.☆11Updated 4 years ago
- Minimal implementation of Denoised Smoothing (https://arxiv.org/abs/2003.01908) in TensorFlow.☆20Updated 3 years ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 5 months ago
- ☆16Updated last year
- Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.☆12Updated last year
- Companion Repo for the Vision Language Modelling YouTube series - https://bit.ly/3PsbsC2 - by Prithivi Da. Open to PRs and collaborations☆14Updated 2 years ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆49Updated 2 years ago
- This project shows how to build a simple handwriting recognizer in Keras with the IAM dataset.☆12Updated 3 years ago
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆14Updated 9 months ago
- FID computation in Jax/Flax.☆24Updated 3 months ago
- A simple Transformer where the softmax has been replaced with normalization☆18Updated 4 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Updated 2 years ago
- Neural Arithmetic Logic Units by Trask et al.☆12Updated 5 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- Utilities for PyTorch distributed☆23Updated last year