simple python Reinforcement learning environment

I am looking for a Python developer to create a custom reinforcement learning environment for my project. The environment needs to have discrete actions based on my detailed specifications. Ideal skills for this job include proficiency in Python, experience in creating custom environments, and knowledge of reinforcement learning algorithms is not needed. The project requires the following:

- Custom reinforcement learning environment with discrete actions (only 2 actions)
- Detailed specifications for the environment provided
- Proficiency in Python
- Experience in creating custom environments

# Use a python Reinforcement learning library to implement RL environment to train and test a simple model (Neural network)
#Important: code in jax(python) in jitted functions
# RL algorithm will be provided in library, no need code algo
#link to a public github RL library with example environments will be shared later.

# simple RL environment in jax(python) to classify a simple single dimensional sine wave like input data into segments of zeros and ones, with input processed sequencial.
#6-8 features will be defined which needs to be calculated in the environment to get observation for step
#6-8 inputs to network with one output
# small (3k-5k rows) training and test data will be provided
# trained model should give acceptable result on test data for the given algo

# Some Experience in ML/AI/data science is needed
# Experience in jax is good but not necessary if willing to learn on your own
# budget will be fixed but time taken will be flexible and little more time can be allowed to complete, no strict deadline