https://github.com/hmunachi/nanodl
A Jax-based library for designing and training transformer models from scratch.
attention
attention-mechanism
deep-learning
distributed-training
flax
gpt
jax
llama
machine-learning
mistral
nlp
transformer
Added: about 1 year ago - Last Synced: 11 months ago
- Created: August 22, 2023

https://github.com/ml4wifi-devs/mapc-sim
IEEE 802.11 MAPC (C-SR) simulator
coordinated-spatial-reuse
jax
simulator
Added: over 1 year ago - Last Synced: 11 months ago
- Created: January 04, 2024
- Relevant topics? true
- External users? true
- Open source license? true
- Active? true
- Fork? false
- Main Language: Python
- Commits: 25
- Committers: 4
- Issues: 0
- Pull Requests: 4
- Owner: ml4wifi-devs
- Stars: 0
- Forks: 0
- Packages: 0

https://github.com/google/trax
Trax β Deep Learning with Clear Code and Speed
deep-learning
deep-reinforcement-learning
jax
machine-learning
numpy
reinforcement-learning
transformer
Added: over 1 year ago - Last Synced: 11 months ago
- Created: October 05, 2019

https://github.com/qazalbash/jaxtro
π An open-source JAX-based gravitational-wave population inference π
astrophysics
gravitational-waves
inference
jax
physics
Added: over 1 year ago - Last Synced: 11 months ago
- Created: December 24, 2023

https://github.com/swanhubx/swanlab
π§SwanLab: track and visualize all the pieces of your machine learning pipeline. θ·θΈͺδΈε―θ§εδ½ ηζΊε¨ε¦δΉ ε
¨ζ΅η¨
data-science
deep-learning
fastapi
jax
machine-learning
mlops
model-versioning
python
pytorch
tensorboard
tensorflow
tracking
transformers
visualization
Added: over 1 year ago - Last Synced: 11 months ago
- Created: November 24, 2023

https://github.com/ami-iit/adam
adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.
adam-robotics
automatic-differentiation
casadi
jax
motion-planning
numpy
optimization
python
pytorch
reinforcement-learning
rigid-body-dynamics
robotics
urdf
Added: over 1 year ago - Last Synced: 11 months ago
- Created: May 25, 2021

https://github.com/hilanzy/moss
Moss is a Python library for Reinforcement Learning.
atari
dm-haiku
dm-launchpad
envpool
impala
jax
library
marl
pettingzoo
ppo
pypi
reinforcement-learning
rlax
Added: over 1 year ago - Last Synced: 11 months ago
- Created: February 19, 2023

https://github.com/corl-team/xland-minigrid
JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid ποΈ
jax
meta-reinforcement-learning
minigrid
python
reinforcement-learning
xland
Added: over 1 year ago - Last Synced: 11 months ago
- Created: December 01, 2023
