Toy implementations of some popular ML optimizers using Python/JAX
☆44Jun 20, 2021Updated 4 years ago
Alternatives and similar repositories for ML-Optimizers-JAX
Users that are interested in ML-Optimizers-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:
- Similarity Encoder (SimEc) Neural Network Framework for learning low dimensional similarity preserving representations☆17Jun 28, 2020Updated 5 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 4 years ago
- Source code of "What can linearized neural networks actually say about generalization?☆20Oct 21, 2021Updated 4 years ago
- Yet Another Neural Network Library 🤔☆27Feb 23, 2026Updated 2 months ago
- Local Attention - Flax module for Jax☆22May 26, 2021Updated 4 years 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.
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 3 years ago
- ☆10Dec 17, 2020Updated 5 years ago
- Hex encode & decode a string, right from your terminal.☆10Jan 5, 2023Updated 3 years ago
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years ago
- Curated list of awesome deep learning resources for complete noobs like you and myself lol!☆14Oct 5, 2018Updated 7 years ago
- Personal Machine Learning Projects☆15Dec 23, 2017Updated 8 years ago
- Shape Matching Virtual Element Method for direct simulation on CAD models!☆13Feb 12, 2022Updated 4 years ago
- Stanford CS244B Distributed Systems Final Project - Secure Peer-to-Peer File Locker☆12Jan 29, 2015Updated 11 years ago
- NoiseGrad (and its extension NoiseGrad++) is a method to enhance explanations of artificial neural networks by adding noise to model weig…☆22May 11, 2023Updated 2 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- All about the fundamental blocks of TF and JAX!☆280Dec 4, 2021Updated 4 years ago
- This is the game Reversi☆12May 21, 2021Updated 4 years ago
- Code for "Holistic Physics Solver: Learning PDEs in a Unified Spectral-Physical Space"☆24Mar 25, 2026Updated last month
- ☆11May 31, 2023Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- pdfplot is a Python library for easily managing your matplotlib figures as PDF files.☆13Jul 8, 2020Updated 5 years ago
- Code of ICML paper arxiv.org/abs/2302.08105☆14May 4, 2023Updated 3 years ago
- Implementing activation functions from scratch in Tensorflow.☆36Feb 13, 2022Updated 4 years 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.
- Repository of the paper "Imperceptible Adversarial Attacks on Tabular Data" presented at NeurIPS 2019 Workshop on Robust AI in Financial …☆16Nov 9, 2021Updated 4 years ago
- Standalone commandline CLI tool for compiling Triton kernels☆20Sep 13, 2024Updated last year
- Geometric Algebra package for JAX☆62Nov 6, 2021Updated 4 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆50Mar 17, 2026Updated last month
- ☆18Updated this week
- My 6th🥇 place solution for Kaggle Shopee competition.☆27Sep 20, 2022Updated 3 years ago
- A research workbench for developing and testing attacks against large language models, with a focus on prompt injection vulnerabilities a…☆49Apr 27, 2026Updated last week
- Lensing analysis and beyond for CMB data☆13Apr 21, 2025Updated last year
- NLP tool to extract emotional phrase from tweets 🤩☆40Oct 18, 2021Updated 4 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- ☆15Oct 29, 2019Updated 6 years ago
- [NeurIPS 2024 Spotlight] Towards Universal Mesh Movement Networks☆19Jul 16, 2025Updated 9 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆120Jun 5, 2022Updated 3 years ago
- ☆39Jan 13, 2016Updated 10 years ago
- minGPT in JAX☆49Jan 10, 2022Updated 4 years ago
- ☆12Sep 22, 2015Updated 10 years ago
- PyTorch reimplementation of the Smooth ReLU activation function proposed in the paper "Real World Large Scale Recommendation Systems Repr…☆22Apr 13, 2022Updated 4 years ago