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