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
Sorting:
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 2 years ago
- Neural Emulator Architectures in JAX.☆23Feb 20, 2026Updated 2 weeks ago
- ☆36Jul 11, 2017Updated 8 years ago
- Cross Atlas Remapping via Optimal Transport☆12Dec 14, 2023Updated 2 years ago
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆41May 5, 2025Updated 10 months ago
- Project Title: Using Machine Learning to Understand the Topology of Knots. This repository contains my Project Log and Code.☆10Nov 7, 2018Updated 7 years ago
- ☆11Aug 19, 2025Updated 6 months ago
- ☆16Jul 7, 2025Updated 7 months ago
- Efficiently Composable Data Augmentation on the GPU with Jax☆42May 16, 2025Updated 9 months ago
- sEEG software☆17Feb 11, 2026Updated 3 weeks ago
- Awesome list about AI4Polymer☆17Sep 23, 2025Updated 5 months ago
- ☆11Sep 16, 2024Updated last year
- Python tools for working with LAMMPS files☆15Jan 5, 2026Updated 2 months ago
- Evaluating methods for estimating aperiodic activity in electrophysiological data.☆16Sep 24, 2024Updated last year
- CPU Emulator with custom Assembly, instruction-set and soon a small C++ Compiler☆17Updated this week
- LAST: Latent Space Assisted Adaptive Sampling for Protein Trajectories☆13Dec 17, 2022Updated 3 years ago
- experimental port of nervana neon kernels in OpenCL☆11Jul 24, 2016Updated 9 years ago
- ☆20Sep 30, 2025Updated 5 months ago
- Supplementary scripts for Z1+ users☆16Dec 12, 2025Updated 2 months ago
- Kubernetes Watcher☆12Feb 6, 2023Updated 3 years ago
- Simple microcanonical Molecular Dynamics simulation of a Lennard-Jones fluid in a periodic boundary☆10Jan 2, 2018Updated 8 years ago
- Template repo for bootstrapping and maintaining SOPs (Standard operation procedures)☆11Sep 1, 2023Updated 2 years ago
- Examples for pscf (polymer self-consistent field theory)☆10Nov 12, 2021Updated 4 years ago
- ☆17Mar 4, 2025Updated last year
- OMNI-P2x: A universal neural network potential for excited states☆12Feb 26, 2026Updated last week
- Understand what physics/algorithms do transformers learn internally when trained on planetary motion☆36Feb 9, 2026Updated 3 weeks ago
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- ☆10Jan 22, 2021Updated 5 years ago
- Modulated automation of cluster expansion based on atomate2 and Jobflow☆12Feb 28, 2026Updated last week
- 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
- Some useful VMD scrpits☆10May 24, 2016Updated 9 years ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆16Nov 20, 2025Updated 3 months ago
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated 11 months ago
- The repository for the generative pretrained model on polymer generation☆18Sep 26, 2025Updated 5 months ago
- ☆12Jan 10, 2023Updated 3 years ago
- ☆12Feb 15, 2026Updated 2 weeks ago
- Benchmark CPUs and GPUs by running molecular dynamics simulations in LAMMPS☆15Oct 25, 2022Updated 3 years ago
- The integrated tempering sampling as a bias in PLUMED2☆11Jan 15, 2021Updated 5 years ago
- Optimal Turn-Right Trajectory using Pontryagin's Maximum Principle.☆12Jul 6, 2023Updated 2 years ago