Transformer creation from scratch using Jax.
☆20Dec 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:
- ☆36Jul 11, 2017Updated 9 years ago
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 2 years ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- Train text generation model with JavaScript.☆15Jul 14, 2024Updated 2 years ago
- Latest version of IFISS software☆11Dec 27, 2025Updated 8 months 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.
- Training methodologies for autoregressive neural operators/emulators in JAX.☆15Jun 1, 2026Updated 3 months 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☆11Jun 6, 2022Updated 4 years ago
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆14Jun 4, 2026Updated 3 months ago
- A GPU-accelerated differentiable fluid simulator written in JAX.☆11Feb 1, 2021Updated 5 years ago
- A variational auto-encoder (VAE) framework with a new type of prior "Variational Mixture of Posteriors" prior, or VampPrior for short.☆10Apr 7, 2021Updated 5 years ago
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆45Nov 21, 2025Updated 9 months ago
- Cross Atlas Remapping via Optimal Transport☆12Dec 14, 2023Updated 2 years ago
- "Neural Emulator Superiority: When Machine Learning for PDEs Surpasses its Training Data" [NeurIPS 2025]☆17May 21, 2026Updated 3 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆42May 5, 2025Updated last year
- [ICLR 2025] On Evluating the Durability of Safegurads for Open-Weight LLMs☆13Jun 20, 2025Updated last year
- Python version of FastPM☆12Jul 20, 2020Updated 6 years ago
- This repository is the official implementation of Topology-Informed Graph Transformer (Choi et al., GRaM Workshop at ICML 2024).☆12Dec 28, 2024Updated last year
- python interface to tetgen☆10Dec 13, 2023Updated 2 years ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- (BESIDE) ``Bridge” Enhanced Signed Directed Network Embedding☆11Dec 22, 2018Updated 7 years ago
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated last year
- Optimal Turn-Right Trajectory using Pontryagin's Maximum Principle.☆12Jul 6, 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.
- Simple optimal control framework for python☆13Aug 16, 2018Updated 8 years ago
- Solving the Navier-Stokes Equations in Python 🐍 simply using NumPy.☆16Jan 28, 2022Updated 4 years ago
- A repository for Mixed Precision Training in JAX based on Equinox.☆16Nov 2, 2025Updated 10 months ago
- Deep network embedding for graph representation learning in signed networks☆14Jul 8, 2020Updated 6 years ago
- Template repo for bootstrapping and maintaining SOPs (Standard operation procedures)☆11Sep 1, 2023Updated 3 years ago
- Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff☆14May 8, 2025Updated last year
- Use SINDY algorithm to discover a dynamical system from coronavirus data☆12Apr 9, 2024Updated 2 years ago
- A rewrite of my fluid simulation project entirely in Rust☆18Nov 22, 2024Updated last year
- Tools for geometric learning☆12Sep 26, 2025Updated 11 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.
- Differentiable Hydrodynamical Simulations☆17Jul 17, 2026Updated last month
- ☆23Nov 24, 2024Updated last year
- Gibbs sampling for expanded ensembles and replica exchange simulations☆11Jun 1, 2014Updated 12 years ago
- ☆17Mar 12, 2025Updated last year
- ☆11Sep 16, 2024Updated last year
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆77Apr 21, 2026Updated 4 months ago
- sEEG software☆17May 28, 2026Updated 3 months ago