JAX/Flax implimentation of 'Attention Is All You Need' by Vaswani et al. (https://arxiv.org/abs/1706.03762)
β15Aug 16, 2021Updated 5 years ago
Alternatives and similar repositories for vanilla-transformer-jax
Users that are interested in vanilla-transformer-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Nadir: Cutting-edge PyTorch optimizers for simplicity & composability! π₯ππ»β15Jun 15, 2024Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jaxβ11Oct 6, 2020Updated 6 years ago
- β18Aug 17, 2024Updated 2 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.β21Sep 18, 2020Updated 6 years ago
- Keras-like APIs for JAX frameworkβ50Mar 25, 2023Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer β’ AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- [ICML 2024] Official PyTorch implementation of the Vectorized Conditional Neural Field.β18Aug 1, 2024Updated 2 years ago
- Benchmarking of diffusion models for global field reconstruction from sparse observationsβ33Mar 22, 2026Updated 6 months ago
- minimalistic AI library that resembles HF's transformersβ13Dec 31, 2024Updated last year
- β20Aug 6, 2024Updated 2 years ago
- Practicum on Supervised Learning in Function Spacesβ36Feb 17, 2022Updated 4 years ago
- Code and Dataset release of "Carpe Diem: On the Evaluation of World Knowledge in Lifelong Language Models" (NAACL 2024)β10Oct 16, 2024Updated 2 years ago
- β19Updated this week
- Procgen2: A community maintained fork of procgenβ13Aug 3, 2026Updated 2 months ago
- νκ΅μ΄ λ¬Έμ₯ λΆμ μμ€ν BCD-KL-Parserβ10Jun 23, 2020Updated 6 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer β’ AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Impementation of Variational Monte Carlo (VMC) for quantum many-body dynamics using JAX.β74Sep 26, 2025Updated last year
- β14Mar 13, 2024Updated 2 years ago
- Layered distributions using FLAX/JAXβ10Dec 13, 2020Updated 5 years ago
- A JAX implementation of stochastic addition.β13Aug 15, 2022Updated 4 years ago
- 200,000+ Sentences about Donald Trump with political bias labelsβ18Jun 2, 2020Updated 6 years ago
- β13Jul 31, 2023Updated 3 years ago
- TANTE: Time-Adaptive Operator Learning via Neural Taylor Expansionβ18Jul 3, 2026Updated 3 months ago
- An RPG Maker MZ pluginβ13Nov 2, 2023Updated 2 years ago
- Paper introducing jax-cosmoβ13Apr 27, 2023Updated 3 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.
- Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"β61Jun 10, 2024Updated 2 years ago
- JAX implementation of Graph Attention Networksβ13Jan 29, 2022Updated 4 years ago
- A Modular Platform for Streamlining Automated Cryo-FIB Workflowsβ26Mar 5, 2025Updated last year
- A custom node for ComfyUI that compares two images using deep learningβ10Jan 18, 2025Updated last year
- β15Oct 29, 2019Updated 6 years ago
- Pytorch implementation of NASA: NEURAL ARTICULATED SHAPE APPROXIMATIONβ12May 4, 2021Updated 5 years ago
- Precision Knowledge Editing (PKE): A novel method to reduce toxicity in LLMs while preserving performance, with robust evaluations and haβ¦β12Nov 26, 2024Updated last year
- β33Feb 22, 2024Updated 2 years ago
- β14Sep 30, 2021Updated 5 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer β’ AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Speech in Flax/JAXβ14Jul 11, 2022Updated 4 years ago
- The PyTorch implementation of paper "KERMIT: Knowledge Graph Completion of Enhanced Relation Modeling with Inverse Transformation"β16Jul 4, 2025Updated last year
- β12Mar 8, 2025Updated last year
- β11Jan 14, 2022Updated 4 years ago
- An implementation of the transformer quantum state, a multi-purpose model for quantum many-body problemsβ39Sep 8, 2023Updated 3 years ago
- β10Mar 14, 2021Updated 5 years ago
- Implementation of language model papers along with several examples [NOT ALL WRITTEN FROM SCRATCH].β12Oct 2, 2024Updated 2 years ago