Transformer creation from scratch using Jax.
☆19Dec 8, 2024Updated last year
Alternatives and similar repositories for transformers
Users that are interested in transformers are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 2 years ago
- Efficiently Composable Data Augmentation on the GPU with Jax☆42May 16, 2025Updated 11 months ago
- ☆36Jul 11, 2017Updated 8 years ago
- Train text generation model with JavaScript.☆15Jul 14, 2024Updated last year
- Latest version of IFISS software☆11Dec 27, 2025Updated 4 months ago
- 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.
- Training methodologies for autoregressive neural operators/emulators in JAX.☆11Nov 5, 2024Updated last year
- A minimal JAX library for connectivity modelling at scale☆11Apr 17, 2026Updated 3 weeks ago
- 2D Stable Fluids & 3D Stable Fluids using the Fast Fourier Transformation implemented efficiently in Julia.☆16Jan 29, 2022Updated 4 years ago
- A simple pseudo-spectral solver for the Direct Numerical Simulation (DNS) of the 3D Taylor-Green Vortex in the Julia programming language☆10Jun 6, 2022Updated 3 years ago
- A GPU-accelerated differentiable fluid simulator written in JAX.☆11Feb 1, 2021Updated 5 years ago
- Fast Kolmogorov-Arnold Network in JAX, initial experiments☆15May 20, 2024Updated last year
- Cross Atlas Remapping via Optimal Transport☆12Dec 14, 2023Updated 2 years ago
- Optimisation on Diffeomorphisms☆12Feb 17, 2025Updated last year
- ☆16Jul 7, 2025Updated 10 months 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.
- experimental port of nervana neon kernels in OpenCL☆11Jul 24, 2016Updated 9 years ago
- ☆11Jan 22, 2021Updated 5 years ago
- ☆14Feb 18, 2024Updated 2 years ago
- Run Claude Code on Ollama☆37Apr 22, 2025Updated last year
- ☆17Mar 4, 2025Updated last year
- A Python interface for the Dawn WebGPU engine☆19Jan 10, 2026Updated 3 months ago
- Understand what physics/algorithms do transformers learn internally when trained on planetary motion☆42Feb 9, 2026Updated 3 months ago
- Python version of FastPM☆12Jul 20, 2020Updated 5 years ago
- ☆23Aug 4, 2025Updated 9 months 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.
- ☆20Sep 30, 2025Updated 7 months ago
- Implementation of various equivariant models in JAX☆19Apr 12, 2024Updated 2 years ago
- Simple optimal control framework for python☆13Aug 16, 2018Updated 7 years ago
- OMNI-P2x: A universal neural network potential for excited states☆13Mar 19, 2026Updated last month
- Dual Octree Graph Networks for Learning Adaptive Volumetric Shape Representations [SIGGRAPH 2022]☆23Dec 6, 2025Updated 5 months ago
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated last year
- Use SINDY algorithm to discover a dynamical system from coronavirus data☆12Apr 9, 2024Updated 2 years ago
- python interface to tetgen☆10Dec 13, 2023Updated 2 years ago
- MSPaint for marimo and other Python notebooks☆25Oct 24, 2025Updated 6 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff☆14May 8, 2025Updated last year
- Indonesian speech/phoneme recognizer powered by Kaldi 2.0 (lhotse, icefall, sherpa).☆15Jun 30, 2023Updated 2 years ago
- Solving the Navier-Stokes Equations in Python 🐍 simply using NumPy.☆16Jan 28, 2022Updated 4 years ago
- This repository contains the codes of "A Lip Sync Expert Is All You Need for Speech to Lip Generation In the Wild", published at ACM Mult…☆15Nov 11, 2022Updated 3 years ago
- Deep network embedding for graph representation learning in signed networks☆14Jul 8, 2020Updated 5 years ago
- Open source web based EEG visualizer and analyzer☆30Mar 10, 2026Updated last month