A JAX nn library
☆21Sep 9, 2025Updated 5 months ago
Alternatives and similar repositories for jaxtorch
Users that are interested in jaxtorch are comparing it to the libraries listed below
Sorting:
- ☆21Mar 15, 2023Updated 2 years ago
- 🖼️📊☆11Jun 9, 2020Updated 5 years ago
- Code publication to the paper "Normalized Attention Without Probability Cage"☆17Nov 9, 2021Updated 4 years ago
- Contrastive Language-Image Pretraining☆144Sep 6, 2022Updated 3 years ago
- ☆22May 3, 2022Updated 3 years ago
- ☆22Jul 15, 2025Updated 7 months ago
- Suite of 500 procedurally-generated NLP tasks to study language model adaptability☆21Jul 16, 2022Updated 3 years ago
- GPT2 Byte Pair Encoding implementation in Golang☆24Jul 9, 2025Updated 7 months ago
- jax version of clip guided diffusion scripts☆90Jan 11, 2024Updated 2 years ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆48Nov 30, 2021Updated 4 years ago
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 3 years ago
- ☆23Jun 18, 2024Updated last year
- Code for paper "Do Language Models Have Beliefs? Methods for Detecting, Updating, and Visualizing Model Beliefs"☆28May 2, 2022Updated 3 years ago
- Official repository of Sparse ISO-FLOP Transformations for Maximizing Training Efficiency☆25Jul 31, 2024Updated last year
- ☆24Jun 4, 2024Updated last year
- ☆63Mar 4, 2022Updated 4 years ago
- ☆65Nov 4, 2021Updated 4 years ago
- Unofficial PyTorch implementation of "Step-unrolled Denoising Autoencoders for Text Generation"☆24Nov 19, 2022Updated 3 years ago
- 🧪 Model Loader API☆33Feb 26, 2024Updated 2 years ago
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…☆25Aug 16, 2023Updated 2 years ago
- ☆30Nov 25, 2021Updated 4 years ago
- Finetune the 1.4B latent diffusion text2img-large checkpoint from CompVis using deepspeed. (work-in-progress)☆36Apr 17, 2022Updated 3 years ago
- ☆34Oct 5, 2020Updated 5 years ago
- Hidden Engrams: Long Term Memory for Transformer Model Inference☆35Jun 26, 2021Updated 4 years ago
- A python algorithm to change the pitch of the voice in real time☆13Dec 13, 2020Updated 5 years ago
- Deep learning based prediction of pi-helices in protein sequences☆11Mar 24, 2023Updated 2 years ago
- [ICML 2024] Official Repository for the paper "Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models"☆10Jul 19, 2024Updated last year
- 💧 Inland water systems are essential to our environment because they are vital ecosystems that are bio-diverse. Thus, finding innovative…☆11Apr 26, 2020Updated 5 years ago
- v objective diffusion inference code for JAX.☆215Apr 14, 2022Updated 3 years ago
- ☆12Jan 10, 2026Updated last month
- Contrastive Language-Audio Pretraining☆87Mar 6, 2022Updated 3 years ago
- codebase for the SIMAT dataset and evaluation☆38Feb 16, 2022Updated 4 years ago
- Official implementation of the ICML2021 paper "Elastic Graph Neural Networks"☆42Jul 25, 2021Updated 4 years ago
- Pedagogical codebase for a simplified score-based generative model design, with training loop☆40Aug 28, 2021Updated 4 years ago
- Tensorflow 2 implementations of the C-SimCLR and C-BYOL self-supervised visual representation methods from "Compressive Visual Representa…☆37Jan 18, 2022Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Jul 19, 2025Updated 7 months ago
- Browser extension for toggling fullscreen on any website.☆10Feb 9, 2020Updated 6 years ago
- Lipid identification software for discovery LC-MS/MS☆12Mar 29, 2019Updated 6 years ago
- Convert Python scripts into a single line of code (oneliner).☆13Jan 23, 2026Updated last month