arition / torch-js
Node.js binding for PyTorch.
☆17Updated 7 months ago
Related projects ⓘ
Alternatives and complementary repositories for torch-js
- Compiling machine learning to wasm and WebGPU☆82Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆45Updated 2 years ago
- VERY MUCH A WORK IN PROGRESS & ACTIVE RESEARCH PROJECT -- DO NOT USE.☆57Updated 7 years ago
- ☆18Updated last year
- 🧠⚙️ Web Neural Network API polyfill based on TensorFlow.js☆102Updated last month
- Deep learning for the Web☆36Updated 3 years ago
- implementation of transformers in tensorflow.js☆28Updated 3 years ago
- ☆12Updated 3 years ago
- Fast matrix-matrix multiplication on web browser using WebGPU☆116Updated 2 months ago
- TensorflowTTS in Tensorflow.js☆17Updated 3 years ago
- ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.☆56Updated 2 years ago
- A generative modelling toolkit for PyTorch.☆70Updated 3 years ago
- Structured inference with Llama 2 in your browser☆52Updated 3 weeks ago
- RWKV (Receptance Weighted Key Value) is a RNN with Transformer-level performance☆36Updated last year
- Contrastive Language-Image Pretraining☆143Updated 2 years ago
- Latent Diffusion Language Models☆67Updated last year
- Another attempt at a long-context / efficient transformer by me☆37Updated 2 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- Nod.ai 🦈 version of 👻 . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository …☆107Updated this week
- Implementation of a Transformer, but completely in Triton☆248Updated 2 years ago
- Save the Tensorflow.js models to the filesystem of ANY Node environment☆7Updated 2 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆234Updated last year
- A case study of efficient training of large language models using commodity hardware.☆68Updated 2 years ago
- ☆38Updated last year
- Evaluate your models with A/B test experiments☆14Updated last year
- webassembly binding for Hora Approximate Nearest Neighbor Search Library☆54Updated 3 years ago
- Serverless APIs on remote GPUs☆56Updated last year
- Babysit your preemptible TPUs☆84Updated last year
- ☆57Updated 2 years ago