A JAX nn library
☆21Sep 9, 2025Updated 7 months ago
Alternatives and similar repositories for jaxtorch
Users that are interested in jaxtorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆21Mar 15, 2023Updated 3 years ago
- 🖼️📊☆11Jun 9, 2020Updated 5 years ago
- ☆13Aug 20, 2021Updated 4 years ago
- ☆62Mar 4, 2022Updated 4 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Apr 6, 2022Updated 4 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆30Nov 25, 2021Updated 4 years ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆48Nov 30, 2021Updated 4 years ago
- trying to make WebGPU a bit easier to use☆19Jan 9, 2024Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆59Jun 23, 2021Updated 4 years ago
- Implementation for <Understanding Robust Overftting of Adversarial Training and Beyond> in ICML'22.☆13Jul 1, 2022Updated 3 years ago
- Official code repository for Instance Selection for GANs.☆44Jan 22, 2021Updated 5 years ago
- jax version of clip guided diffusion scripts☆90Jan 11, 2024Updated 2 years ago
- An implementation of 2021 paper by Geoffrey Hinton: "How to represent part-whole hierarchies in a neural network" in Pytorch.☆57Mar 29, 2021Updated 5 years ago
- GPT2 Byte Pair Encoding implementation in Golang☆25Jul 9, 2025Updated 9 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Jul 19, 2025Updated 8 months ago
- ☆23Jun 18, 2024Updated last year
- ☆32Jan 7, 2024Updated 2 years ago
- JAX bindings for Flash Attention v2☆104Feb 28, 2026Updated last month
- CLOOB Conditioned Latent Diffusion training and inference code☆111Apr 15, 2022Updated 3 years ago
- ☆65Nov 4, 2021Updated 4 years ago
- Training simple models to predict CLIP image embeddings from text embeddings, and vice versa.☆60Mar 31, 2022Updated 4 years ago
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆19Oct 9, 2022Updated 3 years ago
- Hidden Engrams: Long Term Memory for Transformer Model Inference☆35Jun 26, 2021Updated 4 years ago
- Serverless GPU API endpoints on Runpod - Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Companion repository to "Prompt Compression and Contrastive Conditioning for Controllability and Toxicity Reduction in Language Models"☆14May 31, 2023Updated 2 years ago
- Visual search interface☆11Nov 30, 2021Updated 4 years ago
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Jan 16, 2022Updated 4 years ago
- Datasets for Drug Discovery and Development☆10Aug 22, 2020Updated 5 years ago
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 3 years ago
- JAX implementation of Graph Attention Networks☆13Jan 29, 2022Updated 4 years ago
- Train very large language models in Jax.☆209Oct 21, 2023Updated 2 years ago
- OPUS-Rota4: A Gradient-Based Protein Side-Chain Modeling Framework Assisted by Deep Learning-Based Predictors☆11Apr 14, 2022Updated 4 years ago
- Image restoration with neural networks but without learning.☆46May 12, 2022Updated 3 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- A collection of pretrained models for StyleGAN3☆298Jul 18, 2022Updated 3 years ago
- Official implementation of the ICML2021 paper "Elastic Graph Neural Networks"☆42Jul 25, 2021Updated 4 years ago
- Very deep VAEs in JAX/Flax☆46Jun 16, 2021Updated 4 years ago
- MelGAN-VC: Voice Conversion and Audio Style Transfer on arbitrarily long samples using Spectrograms☆11Nov 25, 2021Updated 4 years ago
- ☆12Feb 4, 2025Updated last year
- Dissimilarity Matrix and Sounds from Timbre Space Representation of a Subtractive Synthesizer (Timbre, 2020)☆12Dec 17, 2021Updated 4 years ago
- A CLI tool for using GLIDE to generate images from text.☆66May 5, 2022Updated 3 years ago