JAX/Flax implimentation of 'Attention Is All You Need' by Vaswani et al. (https://arxiv.org/abs/1706.03762)
β15Aug 16, 2021Updated 4 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! π₯ππ»β14Jun 15, 2024Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jaxβ11Oct 6, 2020Updated 5 years ago
- β18Aug 17, 2024Updated last year
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.β21Sep 18, 2020Updated 5 years ago
- Keras-like APIs for JAX frameworkβ50Mar 25, 2023Updated 3 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.
- [ICML 2024] Official PyTorch implementation of the Vectorized Conditional Neural Field.β18Aug 1, 2024Updated last year
- Benchmarking of diffusion models for global field reconstruction from sparse observationsβ33Mar 22, 2026Updated 4 months ago
- β11Nov 4, 2012Updated 13 years ago
- β18Aug 6, 2024Updated last year
- 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 last year
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.β14Dec 21, 2021Updated 4 years ago
- Procgen2: A community maintained fork of procgenβ12Aug 25, 2022Updated 3 years ago
- νκ΅μ΄ λ¬Έμ₯ λΆμ μμ€ν BCD-KL-Parserβ10Jun 23, 2020Updated 6 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits β’ AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Deep NLP 2 (2019.3-5)β10Feb 19, 2019Updated 7 years ago
- Staggered HARd-constrained Physics-Informed Neural Networks for Phase-field modelling of corrosionβ16Feb 19, 2025Updated last year
- β14Mar 13, 2024Updated 2 years ago
- Layered distributions using FLAX/JAXβ10Dec 13, 2020Updated 5 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variationβ12May 24, 2021Updated 5 years ago
- A JAX implementation of stochastic addition.β13Aug 15, 2022Updated 3 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
- Paper introducing jax-cosmoβ13Apr 27, 2023Updated 3 years ago
- JAX implementation of Graph Attention Networksβ13Jan 29, 2022Updated 4 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.
- β23Feb 9, 2025Updated last year
- β15Oct 29, 2019Updated 6 years ago
- Repository containing code for the NAACL 2021 paper (Incorporating External Knowledge to Enhance Tabular Reasoning)β16Jun 20, 2021Updated 5 years ago
- β14Sep 30, 2021Updated 4 years ago
- β10Mar 14, 2021Updated 5 years ago
- Code for the blog postβ12Jan 15, 2021Updated 5 years ago
- β11Jan 14, 2022Updated 4 years ago
- Implementation of language model papers along with several examples [NOT ALL WRITTEN FROM SCRATCH].β12Oct 2, 2024Updated last year
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulationβ117Dec 13, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient β’ AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- [ICML 2026] Official PyTorch implementation of the SMART model.β20May 18, 2026Updated 2 months ago
- Code repository for the paper on "Predicting the Performance of Black-Box LLMs through Self-Queries".β12Jan 9, 2025Updated last year
- Implementation of ICML2023 paper : Coarse-to-Fine: a Hierarchical Diffusion Model for Molecule Generation in 3Dβ51Jul 17, 2023Updated 3 years ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Frameworkβ80Oct 29, 2025Updated 9 months ago
- Evaluation of Sentence Representations in Polishβ22Dec 29, 2022Updated 3 years ago
- β11May 3, 2023Updated 3 years ago
- a character-ai like UI for LLMβ10Dec 3, 2024Updated last year